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

704、二分查找

 题目描述

题目链接：

​​​​​​力扣（LeetCode）官网 - 全球极客挚爱的技术成长平台​​​​​​

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

思路

既然数组是有序的，无论升序还是降序，都可以考虑使用二分查找方法来做，每次砍掉一半的数据量，复杂度O(logn)，比较简单。

代码实现

左闭右闭写法

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

    }
};

左闭右开写法

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

    }
};

27、移除元素

题目描述

题目链接：力扣（LeetCode）官网 - 全球极客挚爱的技术成长平台

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

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

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

 思路：暴力

首先考虑暴力解法：即遍历数组，每查到一个val值，就将后面的元素依次前移，覆盖掉当前元素，同时数组长度减一。比较简单。

代码实现：暴力

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int n = nums.size();
        int i = 0;
        while(i < n){
            if(nums[i] == val){
                for(int j = i; j < n - 1; j++){
                    nums[j] = nums[j + 1];
                }
                n--;
            }
            else i++;
        }
        return n;
    }
};

思路：双指针

题目要求在原地修改数组，其实跟另外创造一个数组，将除val值以外的元素复制到新数组是一样的。若是创造新数组，则需要两个指针分别指向原数组和新数组的赋值位置。而原地修改数组，则需要两个指针分别指向当前位置（右指针、原数组）和即将赋值的位置（左指针、新数组），类比前者复制数组的操作方式。

每一步首先判断右指针指向的元素是否等于val，若相等则忽略，右指针右移，若不等则赋值，左右指针同时右移 ‘nums[left++] = nums[right++]’。如此这般，当右指针遍历完整个数组，左指针则代表新数组的长度，[0, left)区间内没有val值。

代码实现：双指针

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