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

704. 二分查找

二分法，区间的定义一般为两种，左闭右闭即[left, right]，或者左闭右开即[left, right)

示例1

输入: nums = [-1,0,3,5,9,12], target = 9     
输出: 4       
解释: 9 出现在 nums 中并且下标为 4 

示例2

输入: nums = [-1,0,3,5,9,12], target = 2     
输出: -1        
解释: 2 不存在 nums 中因此返回 -1  

前提条件

• 数组为有序数组
• 数组中无重复元素

使用[left, right]要包含right

C++实现（[left, right]）

class Solution {
public:
    int search(vector<int>& nums, int target) {
        sort(nums.begin(), nums.end());
        int left = 0;
        int right = nums.size() - 1; //左闭右闭区间，包含右边界
        while(left <= right){
            int middle = (left + right)/2;
            if(nums[middle] > target){
                right = middle - 1;  //包含右边界，middle已经处理过了，不能包含middle
            }
            else if(nums[middle] < target)
                left = middle + 1;
            else
                return middle;
        }
        return -1;
    }
};

使用[left, right)不包含right

C++实现（[left, right)）

class Solution {
public:
    int search(vector<int>& nums, int target) {
        sort(nums.begin(), nums.end());
        int left = 0;
        int right = nums.size(); //左闭右开区间，不包含右边界
        while(left < right){
            int middle = (left + right)/2;
            if(nums[middle] > target){
                right = middle;    //不包含右边界，所以可以等于middle
            }
            else if(nums[middle] < target)
                left = middle + 1;
            else
                return middle;
        }
        return -1;
    }
};

Python实现（[left, right]）

def search(nums: List[int], target: int) -> int:
    # 左闭右闭区间
    left, right = 0, len(nums) - 1
    # 左闭右闭区间
    while left <= right:
        middle = (left + right) // 2
        if nums[middle] < target:
            left = middle + 1
        elif nums[middle] > target:
            # 左闭右闭区间
            right = middle - 1
        else:
            return middle
    return -1

Python实现（[left, right)）

def search(nums: List[int], target: int) -> int:
    # 左闭右开区间
    left, right = 0, len(nums) - 1
    # 左闭右开区间
    while left < right:
        middle = (left + right) // 2
        if nums[middle] < target:
            left = middle + 1
        elif nums[middle] > target:
            # 左闭右开区间
            right = middle
        else:
            return middle
    return -1

总结

• 注意区间开闭，
• 是否包含注意循环条件

27. 移除元素

题目要求

• 给你一个数组 nums 和一个值 val，你需要 原地 移除所有数值等于 val 的元素，并返回移除后数组的新长度。
• 不要使用额外的数组空间，你必须仅使用 O(1) 额外空间并原地修改输入数组。
• 元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

示例

输入：nums = [3,2,2,3], val = 3
输出：2, nums = [2,2]
解释：函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。你不需要考虑数组中超出新长度后面的元素。例如，函数返回的新长度为 2 ，而 nums = [2,2,3,3] 或 nums = [2,2,0,0]，也会被视作正确答案

思路

因为数组的底层是使用连续的存储空间进行存储的，删除元素本质上是使用目标元素后面的元素覆盖目标元素，不是真正意义上的删除

解法一：暴力解法

这个题目暴力的解法就是两层for循环，一个for循环遍历数组元素 ，第二个for循环更新数组。时间复杂度为O(n)

图片来源

C++实现

class Solution{
public:
    int removeElement(vector<int> &nums, int val){
        int size = nums.size();
        for (int i = 0; i < size; ++i) {
            if(nums[i] == val) {
                for (int j = i + 1; j < size; ++j) {  //最后一个是目标元素，不会进行覆盖，直接元素数目减一
                    nums[j - 1] = nums[j];
                }
                i--; //防止有连续的目标元素，向前移动一位
                size--; //覆盖一个后，元素的数量减一
            }
        }
        return size;
    }
};

解法二：双指针

方法

双指针法（快慢指针法）： 通过一个快指针和慢指针在一个for循环下完成两个for循环的工作。

定义快慢指针

• 快指针：寻找新数组的元素 ，新数组就是不含有目标元素的数组
• 慢指针：指向更新 新数组下标的位置

删除过程如下

图片来源

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;
    }
};

Note： 最后slowIndex就是最后数组的大小。