代码随想录day01 | 704. 二分查找、27. 移除元素

704. 二分查找

思路

定义数组第一位下标为left，最后一位下标为right，数组中间元素为nums[mid]。
循环比较数组中间的值num[mid]和目标值target的值，因为是升序且元素不重复数组，如果nums[mid]<target，那么接下来在mid到right的区间中查找；如果nums[mid]>target，那么接下来在left到mid的区间中查找；如果nums[mid]=target，target的下标就是mid；如果循环结束还没有找到，就返回-1。

代码

写法1（左闭右闭区间）：

class Solution {
    public int search(int[] nums, int target) {
        int left = 0;
        int right = nums.length - 1; //这种定义方式nums[right]是数组最后一位，所以是左闭右闭区间[left,right]
        while(left <= right) { //当left==right，区间[left, right]依然有效，所以用 <=
            int mid = left + (right - left) / 2; // 防止left+right溢出 等同于(left + right)/2
            if (nums[mid] < target){ // target 在右区间，所以下一轮循环应该在[mid + 1, right]区间查找
                left = mid + 1;
            }else if (nums[mid] > target){ // target 在左区间，所以下一轮循环应该在[left, mid - 1]区间查找
                right = mid - 1;
            }else { // nums[mid] == target 找到目标值，返回下标
                return mid;
            }
        }
        // 未找到目标值
        return -1;
    }
}

写法2（左闭右开区间）：

class Solution {
    public int search(int[] nums, int target) {
        int left = 0;
        int right = nums.length; // nums[right - 1]才是数组最后一位,而nums[right]没有值，实际已经越界了，可以看做左闭右开区间[left,right)
        while(left < right) { // 在区间[left,right)中，left==right没有意义，所以是 <
            int mid = left + (right - left) / 2;
            if (nums[mid] < target){
                left = mid + 1;
            }else if (nums[mid] > target){
                right = mid; // 因为nums[mid]大于target，所以可以直接将mid的值赋值为下一次循环时的右边界
            }else {
                return mid;
            }
        }
        return -1;
    }
}

总结

个人认为二分查找左闭右闭的写法比较好理解，而对于左闭右开区间的写法，left等于right没意义，故while循环条件要使用<而不是<=。另一方面，nums[mid] > target时，因为当前nums[mid]不等于target，去左区间继续寻找，而寻找区间是左闭右开区间，所以right更新为mid。

27. 移除元素

思路

思路一，双指针（快慢指针）：
由于题目要求空间复杂度为O(1)，且移除掉元素之后的新数组长度必小于原数组，那么我们可以直接在原数组里修改：设置两个下标变量分别为left=0和right=0，left指向下一个要赋值的位置，right指向当前要处理的位置；
当right<数组长度时，循环向后移动，如果nums[right] = val，不做处理，right继续向后移动，如果nums[right] != val，就把nums[right]赋值给nums[left]，并将left和right都向后移一位。
这种方法简单粗暴，但最坏情况下两个指针会各遍历一遍数组。
思路二，双指针优化：
由于题目并不限制数组输出顺序，那么我们可以使双指针各自从头尾向中间移动，当nums[left] = val时，将nums[right]的值赋值给nums[left]，覆盖原来的var，并使right–，当nums[left] != val时，left++。
这样做的好处是头尾两个指针加起来只需要遍历一遍数组。

代码

class Solution {
    public int removeElement(int[] nums, int val) {
        int left = 0;
        int right = nums.length;
        while(left < right){
            if(nums[right - 1] == val){ //提前判断nums[right]是否为val，是的话直接左移right
                right--;
            }else if(nums[left] == val){ //nums[left] == val时，此元素要移除，可以直接用数组最后一个元素覆盖掉
                nums[left] = nums[right - 1]; //将最后一个元素赋值给nums[left]
                right--; //右指针向左移动
            }else { //nums[left] != val时，此元素要保留，left直接指向下一个元素
                left++;
            }
        }
        return left;
    }
}