代码随想录训练营打卡DAY 1 | 二分法+双指针

代码随想录训练营打卡DAY 1 | 二分法+双指针

二分法

704.二分查找（leetcode链接）

题目描述

给定一个 n 个元素有序的（升序）整型数组 nums 和一个目标值 target ，写一个函数搜索 nums 中的 target，如果目标值存在返回下标，否则返回 -1。

示例1：

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

提示：

• 你可以假设 nums 中的所有元素是不重复的。
• n 将在 [1, 10000]之间。
• nums 的每个元素都将在 [-9999, 9999]之间。

题解

思路：

使用二分法查找元素，需要数组满足顺序存储、有序、元素不重复这3个条件。当满足上述条件时可以考虑使用二分法。
二分法有2种写法，一种为左闭右闭，另一种为左闭右开。笔者习惯前者，故采用第一种写法，即区间为 [left, right]。

核心代码【C++】：

int search(vector<int>& nums, int target) {
        // nums有序且不重复
        int n = nums.size();
        int left = 0, right = n - 1;
        int middle;
        // 二分法，middle指向中间元素
        // 习惯用左闭右闭区间，用<=，因为当left == right时，[left,right]仍有效
        while(left <= right){
            middle = left + ((right - left)/2);//防止溢出
            // 本来写的是：middle = (right + left) / 2;
            /*
            ① 该元素=target，直接返回middle
            ② 该元素<target，[middle + 1,right]
            ③ 该元素>target，[left,middle - 1]
            */
            if(nums[middle] == target) return middle;
            else if(nums[middle] < target) left = middle + 1;
            else right = middle - 1;
            
        }
        return -1;
    }

时间复杂度：O(log n)
空间复杂度：O(1)

双指针

27.移除元素（leetcode链接）

题目描述

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

示例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]，也会被视作正确答案。

示例2：

输入：nums = [0,1,2,2,3,0,4,2], val = 2
输出：5, nums = [0,1,3,0,4]
解释：函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。注意这五个元素可为任意顺序。你不需要考虑数组中超出新长度后面的元素。

提示：

• 0 <= nums.length <= 100
• 0 <= nums[i] <= 50
• 0 <= val <= 100

题解一

思路：

笔者拿到题目的第一想法是使用相向双指针：(暴力解法略)

• 右指针向左遍历直到所指元素 != val
• 左指针向右遍历直到所指元素 == val
• 交换左右指针所指元素，并重复上述操作直至退出循环

核心代码【C++】:

int removeElement(vector<int>& nums, int val) {
        if(nums.size() == 0) return 0;
        // 左指针i，右指针j
        int i = 0, j = nums.size() - 1;
        while(i < j){
            while(i < j && val == nums[j]) j--;// while结束后，nums[j] != val
            while(i < j && val != nums[i]) i++;// while结束后，nums[i] == val
            if(i < j) swap(nums[i],nums[j]);// 这里其实可以用覆盖的方式，但同时就得j--，笔者第一想法是交换，故没有做修改
        }// 此while结束后，i = j
        // 注意临界值
        if(nums[i] != val) return i + 1;
        else return i;
    }
    // 交换，注意参数用“引用”方式传递
    void swap(int& x,int& y){
        int temp = x;
        x = y;
        y = temp;
    }

时间复杂度：O(n)
空间复杂度：O(1)

题解二

思路：

快慢指针，关键在于理解快慢指针分别指什么！

• 快指针含义：新数组当前元素
• 慢指针含义：新数组当前下标

核心代码【C++】：

//代码简洁得我想哭
    int removeElement(vector<int>& nums, int val) {
        // 快慢指针，慢指针slow，快指针fast(搞清楚快慢指针的含义非常重要)
        // 快指针含义：新数组当前元素
        // 慢指针含义：新数组当前下标
        int slow = 0, fast = 0;
        for(fast = 0; fast < nums.size(); fast++){
            if(nums[fast] != val) nums[slow++] = nums[fast];
        }
        return slow;
    }

时间复杂度：O(n)
空间复杂度：O(1)

【参考文档】

1. https://programmercarl.com/0704.%E4%BA%8C%E5%88%86%E6%9F%A5%E6%89%BE.html#%E7%AE%97%E6%B3%95%E5%85%AC%E5%BC%80%E8%AF%BE
2. https://programmercarl.com/0027.%E7%A7%BB%E9%99%A4%E5%85%83%E7%B4%A0.html#%E6%80%9D%E8%B7%AF