代码随想录第1天 |第一章 数组part01

702 二分查找

1、做出假设：

提示：

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

2、输入输出：

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

3、解决方案

二分查找涉及的很多的边界条件，逻辑比较简单，但就是写不好。例如到底是 while(left < right) 还是 while(left <= right)，到底是right = middle呢，还是要right = middle - 1呢？
大家写二分法经常写乱，主要是因为对区间的定义没有想清楚，区间的定义就是不变量。要在二分查找的过程中，保持不变量，就是在while寻找中每一次边界的处理都要坚持根据区间的定义来操作，这就是循环不变量规则。

3.1、数组下标区间左闭右闭

#include <iostream>
#include <vector>

using namespace std;

// 版本一
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;
    }
};


int main() {
    vector<int> nums = { -1,0,3,5,9,12 }; // 给定的有序数组
    int target = 9; // 目标值

    Solution solution; //实例化
    int result = solution.search(nums, target); // 调用solution类中的二分查找函数

    if (result != -1) {
        cout << "Target found at index: " << result << endl;
    }
    else {
        cout << "Target not found in the array." << endl;
    }

    return 0;
}

3.2、数组下标区间左闭右开

#include <iostream>
#include <vector>

using namespace std;


// 版本二
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); // 使用右移操作 >> 1 取代除法 / 2 是一种优化方法
            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;
    }
};


int main() {
    vector<int> nums = { -1,0,3,5,9,12 }; // 给定的有序数组
    int target = 9; // 目标值

    Solution solution; //实例化
    int result = solution.search(nums, target); // 调用solution类中的二分查找函数

    if (result != -1) {
        cout << "Target found at index: " << result << endl;
    }
    else {
        cout << "Target not found in the array." << endl;
    }

    return 0;
}

27. 移除元素

题目假设

示例 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。
注意这五个元素可为任意顺序。你不需要考虑数组中超出新长度后面的元素。

1、暴力解法

#include <iostream>
#include <vector>

using namespace std;

// 时间复杂度：O(n^2)
// 空间复杂度：O(1)
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--; // 因为下标i以后的数值都向前移动了一位，所以i也向前移动一位
                size--; // 此时数组的大小-1
            }
        }
        return size;
    }
};

// 函数用于输出 vector 中的元素
void printVector(const std::vector<int>& vec) {
    std::cout << "nums = [";
    for (size_t i = 0; i < vec.size(); ++i) {
        std::cout << vec[i];
        if (i != vec.size() - 1) { // 添加 ，号
            std::cout << ", ";
        }
    }
    std::cout << "]" << std::endl;
}


int main() {
    vector<int> nums = { 0,1,2,2,3,0,4,2 }; // 给定的有序数组
    int val = 2; // 目标值

    Solution solution; // 实例化
    int result = solution.removeElement(nums, val); // 调用solution类中的二分查找函数
    
    // 调用函数输出 vector 中的元素
    printVector(nums);
    // 输出nums中的元素个数
    cout << "size: " << result << endl;
 
    return 0;
}

2、双指针(快慢指针)

方式一：

// 时间复杂度：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;
    }
};

方式二：

/**
* 相向双指针方法，基于元素顺序可以改变的题目描述改变了元素相对位置，确保了移动最少元素
* 时间复杂度：O(n)
* 空间复杂度：O(1)
*/
class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int leftIndex = 0;
        int rightIndex = nums.size() - 1;
        while (leftIndex <= rightIndex) {
            // 找左边等于val的元素
            while (leftIndex <= rightIndex && nums[leftIndex] != val) {
                ++leftIndex;
            }
            // 找右边不等于val的元素
            while (leftIndex <= rightIndex && nums[rightIndex] == val) {
                --rightIndex;
            }
            // 将右边不等于val的元素覆盖左边等于val的元素
            if (leftIndex < rightIndex) {
                nums[leftIndex++] = nums[rightIndex--];
            }
        }
        return leftIndex;   // leftIndex一定指向了最终数组末尾的下一个元素
    }
};