第一章_数组_Part1

二分查找

题目描述

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

示例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]）

• 初始值：left = 0，right = nums.size() - 1

• 查找条件：while(left <= right)，使得[left,right]有意义

• 二分流程

  • nums[mid] > target时，说明target在左区间，由于target一定不等于nums，因此左区间下标结束位置应该设置为mid - 1
  • nums[mid] < target时，说明target在右区间，由于target一定不等于nums，因此右区间下标起始位置应该设置为mid + 1

写法二：左闭右开（[left,right)）

• 初始值：left = 0，right = nums.size()

• 查找条件：while(left < right)，使得[left,right)有意义

• 二分流程

  • nums[mid] > target时，说明target在左区间，寻找区间是左闭右开区间，所以right更新为middle
  • nums[mid] < target时，说明target在右区间，由于target一定不等于nums，因此右区间下标起始位置应该设置为mid + 1

时空复杂度

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

代码实现

测试地址：https://leetcode.cn/problems/binary-search/

写法一：左闭右闭

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

写法二：左闭右开

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

移出元素

题目描述

给你一个数组 nums 和一个值 val，你需要 原地 移除所有数值等于 val 的元素，并返回移除后数组的新长度。

不要使用额外的数组空间，你必须仅使用 O(1) 额外空间并原地修改输入数组。

元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

示例 1: 给定 nums = [3,2,2,3], val = 3, 函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。 你不需要考虑数组中超出新长度后面的元素。

示例 2: 给定 nums = [0,1,2,2,3,0,4,2], val = 2, 函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。

解题思路

由于数组的元素在内存中是连续的，因此在删除某个元素后，只能进行数组元素覆盖的方式实现这个目的。

暴力解法

总体思路：

使用两层for循环，第一层遍历数组元素，找到元素对应下标，第二层循环将当前元素后面的元素进行依次覆盖，实现移出元素的目的。

时空复杂度分析：

• 时间复杂度：O(n^2)

• 空间复杂度：O(1)

双指针

总体思路：

使用快慢指针在一层for循环下，完成暴力解法两层for循环的工作。

其中快指针指向的元素为新数组中包含的元素，即不等于val的元素

慢指针指向的是新数组的下标位置

通过快指针遍历数组元素，只要快指针指向的元素值不等于val，那么同时移动慢指针下标赋值操作，如果快指针的下标恰好指向了val对应的元素，那么跳过慢指针的赋值操作，直至快指针下标继续移动到不为val元素为止。

时空复杂度分析

• 时间复杂度：O(n)

• 空间复杂度：O(1)

代码实现

测试地址：https://leetcode.cn/problems/remove-element/

暴力解法：

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int size = nums.size();
        for (int i = 0; i < size; i++) {
            if (val == nums[i]) { // 发现需要移除的元素，就将数组集体向前移动一位
                for (int j = i + 1; j < size; j++) {
                    nums[j - 1] = nums[j];
                }
                i--; // 因为下标i以后的数值都向前移动了一位，所以i也向前移动一位
                size--;
            }
        }
        return size;
    }
};

双指针：

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int slow = 0;
        for (int fast = 0; fast < nums.size(); fast++) {
            if (nums[fast] != val) {
                nums[slow++] = nums[fast];
            }
        }
        return slow;
    }
};