LeetCode 第一天 704. 二分查找，27. 移除元素

数组理论基础，704. 二分查找，27. 移除元素

704.二分查找

题目:

给定一个 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

提示：

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

题解：

关键词：有序的（升序）整型

二分查找算法仅适用于有序序列。

1. 有序系列
2. [left, right] 指定搜索区域。left为初始，right为最大值
3. 找到 [left, right]的中间元素，下标mid 。
4. 中间是否找到等于。
5. 如果没有找到，中间元素小于目标值，则往右边找。否则往左边找。

以下图解，查找目标值为 31

所有元素的位置下标，分别用 0~9 表示，中间元素的位置为 ⌊ 0 + (9 - 0) / 2 ⌋ = 4，如下图所示：

中间元素 27 < 31，可以断定 [0, 4] 区域内绝对没有 31，目标元素只可能位于 [5, 9] 区域内，如下图所示：

在 [5, 9] 区域内，中间元素的位置为 ⌊ 5 + (9 - 5) / 2 ⌋ = 7，如下图所示：

中间元素 35 > 31，可以断定 [7, 9] 区域内绝对没有 31，目标元素只可能位于 [5,6] 中，如下图所示：

在 [5, 6] 区域内，中间元素的位置为 ⌊ 5 + (6- 5) / 2 ⌋ = 5，中间元素就是 31，成功找到目标元素。

c++代码实现:

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

python 代码实现:

class Solution:
    def search(self, nums: List[int], target: int) -> int:
        l = len(nums)
        left = 0
        right = l - 1
        while left <= right:
            mid = left + (right - left) // 2
            if nums[mid] == target:
                return mid
            if (nums[mid] < target):
                left = mid + 1
            else:
                right = mid - 1
        return -1

疑问:

为什么要 left + (right - left) / 2, 而不直接 (left + right)/2 ?

答： 这道题依据提示范围，可以使用(left+right)/2，不会造成溢出情况。

但如果范围提示很大，那(left+right) 一旦溢出（即大于$2^{32}$), 就无法找出。所以就先减，再除（降低right）。避免溢出。

27.移除元素

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

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

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

说明:

为什么返回数值是整数，但输出的答案是数组呢?

请注意，输入数组是以「引用」方式传递的，这意味着在函数里修改输入数组对于调用者是可见的。

你可以想象内部操作如下:

// nums 是以“引用”方式传递的。也就是说，不对实参作任何拷贝
int len = removeElement(nums, val);

// 在函数里修改输入数组对于调用者是可见的。
// 根据你的函数返回的长度, 它会打印出数组中 该长度范围内 的所有元素。
for (int i = 0; i < len; i++) {
    print(nums[i]);
}

示例 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,4,0,3]
解释：函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。注意这五个元素可为任意顺序。你不需要考虑数组中超出新长度后面的元素。

提示：

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

题解:

关键词：不要使用额外的数组空间O(1)，元素的顺序可以改变。

可由上面二分查找思路

如果找到，直接替换等于，然后移动右边替换后的下标。否则就往左加。

c++ 代码实现:

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int size = nums.size() - 1;
        int begin = 0;
        // 从末尾开始替换。相同的替换到末尾。因为题目元素的顺序可以改变
        while ( size >= begin ){
            // 从头遍历，如果等于，就赋值后面的值。因为begin还没加，所以就算赋值了，也还会从当前赋值的值，开始判断是否等于。
            if (nums[begin] == val){
                nums[begin] = nums[size];
                --size;
            }else{
                ++begin;
            }
        }
        return begin;
    }
};

python 代码实现:

class Solution:
    def removeElement(self, nums: List[int], val: int) -> int:
        end = len(nums) - 1
        begin = 0
        while end >= begin :
            if nums[begin] == val:
                nums[begin] = nums[end]
                end -= 1
            else:
                begin += 1
        return begin