Follow Carl To Grow|【LeetCode】704. 二分查找，27. 移除元素

【LeetCode】704. 二分查找

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

思考：指明了使用二分法，关键是考虑左右边界的变化。

代码A：

class Solution {
public:
    int search(vector<int>& nums, int target) {
        // 左闭右闭区间，为了保证循环合法，while应该<=，判断时都要去掉mid
        int mid = 0, left = 0, right = nums.size() - 1;
        while (left <= right)
        {
            mid = (left + right) / 2;
            if (target < nums[mid])
            {
                right = mid - 1;
            }
            else if (target > nums[mid])
            {
                left = mid + 1;
            }
            else
            {
                return mid;
            }
        }
        return -1;
    }
};

代码B：

class Solution {
public:
    int search(vector<int>& nums, int target) {
        // 左闭右开区间，为了保证循环合法，while应该<，判断时注意左边界去掉mid
        int mid = 0, left = 0, right = nums.size();
        while (left < right)
        {
            mid = (left + right) / 2;
            if (target > nums[mid])
            {
                left = mid + 1;
            }
            else if (target < nums[mid])
            {
                right = mid;
            }
            else
            {
                return mid;
            }
        }
        return -1;
    }
};

---

【LeetCode】27. 移除元素

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

思路：第一反应是直接遍历调用erase，时间复杂度O(n²)。仔细想想还能用双指针，时间复杂度O(n)。

代码A：

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        for (vector<int>::iterator it = nums.begin(); it != nums.end();)
        {
            if (*it == val)
            {
            	// 避免迭代器失效
                it = nums.erase(it);
            }
            else
            {
                ++it;
            }
        }
        return nums.size();    
    }
};

代码B：

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];
                ++slow;
            }
        }
        return slow;    
    }
};

---

心态：“第一章 数组part01” 拿下！
参考资料