【CV炼丹师勇闯力扣训练营 Day1】

CV炼丹师勇闯力扣训练营

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

---

一、704 二分查找

给定一个 n 个元素有序的（升序）整型数组 nums 和一个目标值 target ，写一个函数搜索 nums 中的 target，如果目标值存在返回下标，否则返回 -1。
【注意】使用二分查找的前提：有序数组、且无重复元素
【难点】边界问题，要保证左闭右闭区、左闭右开两种区间定义始终有意义，即确保不变量始终成立

代码如下（C++）：

// 左闭右闭
class Solution {
public:
    int search(vector<int>& nums, int target) {
        int left = 0;
        int right = nums.size() - 1; // 定义target在左闭右闭的区间里，[left, right]
        while (left <= right) { // 当left==right，区间[left, right]依然有效，所以用 <=
            int middle = left + ((right - left) / 2);// 防止溢出 等同于(left + right)/2
            if (nums[middle] > target) {
                right = middle - 1; // target 在左区间，所以[left, middle - 1]
            } else if (nums[middle] < target) {
                left = middle + 1; // target 在右区间，所以[middle + 1, right]
            } else { // nums[middle] == target
                return middle; // 数组中找到目标值，直接返回下标
            }
        }
        // 未找到目标值
        return -1;
    }
};

// 左闭右开
class Solution {
public:
    int search(vector<int>& nums, int target) {
        int left = 0;
        int right = nums.size(); // 定义target在左闭右开的区间里，即：[left, right)
        while (left < right) { // 因为left == right的时候，在[left, right)是无效的空间，所以使用 <
            int middle = left + ((right - left) >> 1);
            if (nums[middle] > target) {
                right = middle; // target 在左区间，在[left, middle)中
            } else if (nums[middle] < target) {
                left = middle + 1; // target 在右区间，在[middle + 1, right)中
            } else { // nums[middle] == target
                return middle; // 数组中找到目标值，直接返回下标
            }
        }
        // 未找到目标值
        return -1;
    }
};

代码如下（Python）：

# 左闭右闭
class Solution:
    def search(self, nums, tar) -> int:
        left, right = 0, len(nums) - 1
        while(left <= right):
            mid = left + ((right - left) >> 1)

            if(nums[mid] > tar):
                right = mid - 1
            elif(nums[mid] < tar):
                left = mid + 1
            else:
                return mid
        return -1

# 左闭右开
class Solution:
    def search(self, nums, tar):
        left, right = 0, len(nums)
        while(left <= right):
            mid = left + ((right - left) >> 1)
            
            if(nums[mid] > tar):
                right = mid  # 右开不取值，所以下一次循环right为mid
            elif(nums[mid] < tar):
                left = mid + 1  # 左闭取值，所以下一次循环left不能算上mid,得从mid+1开始
            else:
                return mid
        return -1

二、27 移除元素

给你一个数组 nums 和一个值 val，你需要 原地 移除所有数值等于 val 的元素。元素的顺序可能发生改变。然后返回 nums 中与 val 不同的元素的数量。

假设 nums 中不等于 val 的元素数量为 k，要通过此题，您需要执行以下操作：

• 更改 nums 数组，使 nums 的前 k 个元素包含不等于 val 的元素。nums 的其余元素和 nums 的大小并不重要。
• 返回 k

暴力解法：两层for循环，第⼀个for循环遍历数组元素 ，第⼆个for循环更新数组

☆ 双指针法（快慢指针法）： 通过⼀个快指针和慢指针在⼀个for循环下完成两个for循环的⼯作。
快指针：寻找新数组的元素 ，新数组就是不含有⽬标元素的数组
慢指针：指向更新 新数组下标的位置，最终指向新数组尾地址，就是新数组的大小

引用代码随想录中的一张动图，能够直观看出快慢指针的运行流程

代码如下（C++）：

// 时间复杂度：O(n)
// 空间复杂度：O(1)
class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int slowIndex = 0;
        for (int fastIndex = 0; fastIndex < nums.size(); fastIndex++) {
            if (val != nums[fastIndex]) {
                nums[slowIndex++] = nums[fastIndex];
            }
        }
        return slowIndex;
    }
};

代码如下（Python）：

class Solution:
    def removeElement(self, nums, val) -> int:
        # 快慢指针
        fast = 0  # 快指针
        slow = 0  # 慢指针
        size = len(nums)
        while fast < size:  # 不加等于是因为，a = size 时，nums[a] 会越界
            # slow 用来收集不等于 val 的值，如果 fast 对应值不等于 val，则把它与 slow 替换
            if nums[fast] != val:
                nums[slow] = nums[fast]
                slow += 1  # 【注意】Python中没有++
            fast += 1
        return slow

---

总结

• 二分查找的两种区间
• 用快慢指针来处理数组元素移除问题