代码随想录算法训练营第一天 | 704. 二分查找、27. 移除元素

玛玛哈哈34

已于 2023-06-11 14:41:53 修改

阅读量1k

点赞数 1

文章标签：算法数据结构 leetcode c++

于 2023-06-07 23:01:37 首次发布

本文链接：https://blog.csdn.net/weixin_46108098/article/details/131093195

版权

一、704.二分查找

今天明确了二分查找的两种写法：左闭右闭[left,right]和左闭右开[left,right)，两种写法如下：

1.左闭右闭[left,right]

代码如下：

class Solution {
public:
    int search(vector<int>& nums, int target) {
        int left = 0, right = nums.size() - 1;
        while (left <= right) {
            int mid = left + (right - left)/2;
            if (nums[mid] > target) 
                right = mid - 1;
            else if (nums[mid] < target) 
                left = mid + 1;
            else 
                return mid; 
        }
        return -1;
    }
};

其中的关键点为：
1）while (left <= right)，<=的原因在于，端点是闭区间时，两端点值相同仍然是合法的区间，例如区间[1,1]；
2）right = mid - 1，当nums[mid]大于target时，需要更新左区间的右边界，由于nums[mid]明确大于target，一定不在下一次的搜索区间内，因此右边界更新为mid-1；
3）left = mid + 1，同理，当nums[mid]小于target时，需要更新右区间的左边界，且nums[mid]不在下一次搜索区间内，因此左边界更新为mid+1；
4）right = nums.size() - 1，右边为闭区间，右边界为size-1。

2.左闭右开[left,right）

代码如下：

class Solution {
public:
    int search(vector<int>& nums, int target) {
        int left = 0, right = nums.size();
        while (left < right) {
            int mid = left + (right - left) / 2;
            if (nums[mid] > target)
                right = mid;
            else if (nums[mid] < target)
                left = mid + 1;
            else 
                return mid;
        }
        return -1;
    }
};

其中的关键点为：
1）right = nums.size()，右边为开区间，右边界为size（因为右端点的值是不取到的）；
2）left < right，区间左闭右开，端点值相同区间不合法，例如[1,1)；
3）right = mid，当nums[mid]大于target时，需要更新左区间的右边界，nums[mid]明确不在下一次搜索范围内，但右边本身是开区间，取不到端点值，故右边界更新为mid；
4）left = mid + 1，当nums[mid]小于target时，需要更新右区间的左边界，nums[mid]明确不在下一次搜索范围内，因为左边是闭区间，所以左边界更新为mid+1。

在两种方法中，mid = left + (right - left) / 2，效果等同于mid = (left + right) / 2，但这种取中间值的写法可以防止溢出。也可以写成mid = left +((right - left)>>1)（右移一位相当于除以2），运行速度更快。

注意，mid的定义是在while循环内部，因为每次循环都会更新mid的位置，自己写的时候经常会定义在while外面，以后要多留个心眼。

二、27.移除元素

移除元素可以使用暴力求解，也可以使用双指针的方法。今天新学到了双指针的方法，它可以节省一层for循环要做的事。

1.暴力求解

两层for循环，第一层遍历数组寻找需要移除的元素，第二层将需要移除的元素后面的值向前移一位。

代码如下：

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int result = nums.size();
        for (int i = 0; i < result; i++){
            if (nums[i] == val){//发现需要移除的元素时，将后面的数组集体向前移一位
                for (int j = i + 1; j < result; j++){
                    nums[j-1] = nums[j];
                }
                i--;//因为第二个for循环中把i以后的位置往前移了一位，在移动完后第一个for循环会执行i++，再次开始遍历的位置应该是前一位，所以i也向前移一位
                result--;//数组长度减1
            }
        }
        return result;
    }
};

2.双指针法

定义：
快指针fastIndex，用来寻找新数组（即删除目标后的数组）里需要的元素；
慢指针slowIndex，用来获取新数组中需要更新的位置（即新数组的下标值）。
两个指针都是在同一个数组上进行操作。

代码如下：

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int slowIndex = 0;
        for (int fastIndex = 0; fastIndex < nums.size(); fastIndex++){
            if (nums[fastIndex] != val){//如果快指针指向的不是需要删除的值
                nums[slowIndex] = nums[fastIndex];//慢指针也指向这个值，它是新数组需要的元素
                slowIndex++;//慢指针移动到下一个位置，等待快指针的判断
            }
        }
        return slowIndex;//结束循环后慢指针的值是新数组的长度，如新数组下标为0-4，慢指针指向的为5
    }
};