美文网首页leetcode
704. 二分查找

704. 二分查找

作者: 胖胖的熊管家 | 来源:发表于2023-01-27 22:35 被阅读0次

    读题思路

    首先,题目中明确说明输入的数组为 有序的(升序)整型数组、并且无重复元素,所以如题所示,那自然就是用二分查找法来查找咯。

    解题思路

    二分法:

    1. 找到中间值,中值下标为mid(初始值就是总数/2),查找的总下标为flag(初始值为总数)
    2. 将要查找的值(目标值)与中间值进行比较;根据大小的结果,调整mid值和总下标,还有将下标左移或右移(+1 or -1)
    3. 重复步骤1,2;直至找到目标值 或 中值mid与总下标发生矛盾。

    出现的问题

    情况一(JS):

    var search = function(nums, target) {
        let flagR = nums.length, flagL = 0
        let mid = Math.round((flagR-flagL) / 2) //考虑到奇数除以2会是小数,所以要取整
    
        while(mid > flagL & mid < flagR){//能够继续下去的条件是mid在flag间
            if(nums[mid] == target){ //中值就是目标值 那就皆大欢喜 直接输出
                return mid
            }
            else if(nums[mid] < target){ //因为是升序,所以这里意味着目标值在中值右边,所以flag调整为右端;若为降序则相反
                flagL = mid
                mid = Math.round((flagR-flagL) / 2)
            }
            else if(nums[mid] > target){ //同理,flag调整为左端
                flagR = mid
                mid = Math.round((flagR-flagL) / 2)
            }
            else{
                return -1
            }
            //console.log(mid)
        }
    };
    

    问题:
    mid值的下标并没有如期改变。
    以nums = [-1,0,3,5,9,12],target = 9为例
    第一次迭代(初始化):

    flagL=0, flagR=6
    此时mid=3,nums[3]=5, < target 9,所以符合条件2(第一个else if)

    第二次迭代:

    flagL=3, flagR=6
    \color{red}{此时mid=2,nums[2] = 3 【问题在这!】}

    原因:
    在多次迭代时,mid值没有正确改变。
    解决方法:

    在flag移到右端的情况下,mid的下标需要+=;而左端不需要
    mid = Math.round((flagR-flagL) / 2)
    改为
    mid +=  Math.round((flagR-flagL) / 2) 
    

    在修改过程中发现flag值初始化也有问题,
    应改为flagR = nums.length-1,因为是下标的范围嘛。
    还有return -1的位置也不恰当。

    情况二(JS):

    修改后的代码提交部分成功

    var search = function(nums, target) {
        let flagR = nums.length-1, flagL = 0
        let mid = Math.round((flagR-flagL) / 2) //考虑到奇数除以2会是小数,所以要取整
    
        while(mid > flagL & mid < flagR){ //能够继续下去的条件是mid在flag间
            if(nums[mid] == target){ //中值就是目标值 那就皆大欢喜 直接输出
                return mid
            }
            else if(nums[mid] < target){ //因为是升序,所以这里意味着目标值在中值右边,所以flag调整为右端;若为降序则相反
                flagL = mid
                mid += Math.round((flagR-flagL) / 2)
            }
            else if(nums[mid] > target){ //同理,flag调整为左端
                flagR = mid
                mid = Math.round((flagR-flagL) / 2)
            }      
            console.log(mid)
        }
        return -1
    };
    

    问题:

    报错

    原因:
    极端情况没考虑。
    考虑极端情况:

    1. 如报错图所示,数组长度为1
    2. target在首或尾。所以就是考虑循环边界的问题。

    解决方法:
    首先,对于数组长度为1的情况,主要是flagL和flagR需要移位,flagL = mid+1flagR = mid-1,还有while循环条件改为mid >= flagL & mid <= flagR

    接下来,仍旧报错,是target在首或尾的极端情况。如:nums=[-1,0,5], target=5
    经过分析,问题出在
    flagL=2,flagR=2, mid = 1+Math.round(0.5)时,这个mid其实就不准了。所以要改变mid计算方式。
    \color{green}{经过思考+查阅,mid的计算改为}

    mid = Math.round((flagR+flagL) / 2)
    

    经过优化后,JS的解法为

    var search = function(nums, target) {
        let flagR = nums.length-1, flagL = 0
        // let mid = Math.round((flagR+flagL) / 2) //考虑到奇数除以2会是小数,所以要取整
    
        while(flagL <= flagR){ //能够继续下去的条件是mid在flag间,也就是flagL要小于flagR
            mid = Math.round((flagR+flagL) / 2)
            if(nums[mid] == target){ //中值就是目标值 那就皆大欢喜 直接输出
                return mid
            }
            else if(nums[mid] < target){ //因为是升序,所以这里意味着目标值在中值右边,所以flag调整为右端;若为降序则相反
                flagL = mid+1
            }
            else if(nums[mid] > target){ //同理,flag调整为左端
                flagR = mid-1
            }      
        }
        return -1
    };
    

    优化后,时间也变短了:


    优化前 vs 优化后

    Python解法为

    class Solution(object):
        def search(self, nums, target):
            """
            :type nums: List[int]
            :type target: int
            :rtype: int
            """
            flagL = 0
            flagR = len(nums)-1
            while(flagL<=flagR):
                mid = int((flagL+flagR) / 2)
                if(nums[mid] == target):
                    return mid
                elif(nums[mid] < target):
                    flagL = mid+1
                elif(nums[mid] > target):
                    flagR = mid-1
            return -1
    

    发现用Python无论是时间还是内存都比JS少诶。


    提交结果对比一览

    总结

    今天算是完全自己从头到尾刷了一道简单的LeetCode题,感觉很不错。
    以前一直以为自己对二分法很懂了,现在才发现是纸上谈兵,写起代码来还是有坑的。
    不过这之后应该算是能说对二分法彻底掌握了哈哈哈!

    相关文章

      网友评论

        本文标题:704. 二分查找

        本文链接:https://www.haomeiwen.com/subject/irkehdtx.html