算法记录 | 数组part01

算法记录 | 数组part01

LeetCode 704-二分查找

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

输入: nums = [-1,0,3,5,9,12], target = 9
输出: 4
解释: 9 出现在 nums 中并且下标为 4

题目链接：leetcode.cn/problems/bi…

题解

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

解题思路

有序数组查找首先考虑二分查找，二分查找时间复杂度为O(log n)。这里需要注意区间有两种形式，第一种是左闭右开[left, right) 第二种是左闭右闭[left, right]。它们的区别是右边界的值能不能取。

LeetCode 27-移除元素

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

题目链接：leetcode.cn/problems/re…

markdown
复制代码
输入：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]，也会被视作正确答案

题解

class Solution
{
public:
    int removeElement(vector<int> &nums, int val)
    {
        // 快慢指针
        int fast = 0, slow = 0;
        int size = nums.size();
        for (; fast < size; fast++)
        {
            // 碰到待移除的元素就跳过，然后再移动覆盖掉跳过的元素
            if (nums[fast] != val)
            {
                nums[slow] = nums[fast];
                slow++;
            }
        }
        return slow;
    }
};

解题思路

第一次解题用的是暴力法，之前做过一个移动零的类似题目，使用双指针优化。这里快慢指针的目的是记录放置后面过来元素的下表以及需要被移动元素的下标。