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

Leetcode 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
提示：
你可以假设 nums 中的所有元素是不重复的。
n 将在 [1, 10000]之间。
nums 的每个元素都将在 [-9999, 9999]之间。

思路分析：
题意给出数组nums为有序数组，且不重复元素，则适合使用二分查找。相对于顺序遍历，二分查找时间复杂度更低，为log(n). 二分查找根据查找区间分为2种：左闭右闭合左闭右开。

代码实现：
左闭右闭

class Solution {
public:
    int search(vector<int>& nums, int target) {
        int left = 0, right = nums.size()-1, middle;
        while (left <= right) {   // 左闭右闭区间，left==right是有意义的
            middle = (left+right)/2;
            if (nums[middle] < target) {
                left = middle + 1; // 左闭右闭区间，nums[middle] < target说明target在右侧，右侧第一个index是middle + 1
            } else if (nums[middle] > target) {
                right = middle - 1; // 左闭右闭区间，nums[middle] > target说明target在左侧，左侧第一个index是middle - 1
            } else {
                return middle;
            }
        }
        return -1;
    }
};

左闭右开

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

本题可以拓展出求x的算术平方根
题目：给你一个非负整数 x ，计算并返回 x 的 算术平方根 。由于返回类型是整数，结果只保留 整数部分 ，小数部分将被 舍去 。
注意：不允许使用任何内置指数函数和算符，例如 pow(x, 0.5) 或者 x ** 0.5 。
代码实现

class Solution {
public:
    int mySqrt(int x) {
    int left = 0;
    int right = x;
    int res;
    while (left <= right) {
      int middle = left + (right - left) / 2;
      if ((long long)middle * middle <= x) {
        res = middle;
        left = middle + 1;
      } else {
        right = middle -1;
      }
    }
    return res;
  }
};

Leetcode 27. 移除元素

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

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

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

思路分析：
本题和在链表中删除某元素相似，链表中是找到值为某值的节点，同时记录其上一个节点，用上一节点的next指针直接指向当前节点的下一个节点，即覆盖当前元素。不过区别是链表中除了覆盖外，还可以删除（当然删除导致链表断了），而数组存储地址是连续的，只能采用覆盖。结合链表的解题思路，本题也能想到使用双指针。用索引值slow和fast模拟两个指针，fast遍历nums,如果出现nums[fast]和val不相等，则说明是移除val后的nums数组的其中一个元素，用nums[slow]记录下来，同时更新slow值。因为题意说不需要考虑数组中超出新长度后面的元素，则slow就是新数组nums的长度。

代码实现：

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