Letcode 上的双指针题--方便复习

Posted 2021-11-04 行码阁119

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

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

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

示例 1: 给定 nums = [3,2,2,3], val = 3, 函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。 你不需要考虑数组中超出新长度后面的元素。

示例 2: 给定 nums = [0,1,2,2,3,0,4,2], val = 2, 函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。

解法一 暴力解法-直接用两个for循环来结题  这也是作者最开始想到的办法 

/ 时间复杂度：O(n^2)
// 空间复杂度：O(1)
class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int size = nums.size();
        for (int i = 0; i < size; i++) {
            if (nums[i] == val) { // 发现需要移除的元素，就将数组集体向前移动一位
                for (int j = i + 1; j < size; j++) {
                    nums[j - 1] = nums[j];
                }
                i--; // 因为下表i以后的数值都向前移动了一位，所以i也向前移动一位
                size--; // 此时数组的大小-1
            }
        }
        return size;

    }
};

解法二 双向指针解法（图借鉴于代码随想录 (programmercarl.com)）

这样就不需要利用两个fou 循环反复进行覆盖

 

// 时间复杂度：O(n^2)
// 空间复杂度：O(1)
class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int size = nums.size();
        for (int i = 0; i < size; i++) {
            if (nums[i] == val) { // 发现需要移除的元素，就将数组集体向前移动一位
                for (int j = i + 1; j < size; j++) {
                    nums[j - 1] = nums[j];
                }
                i--; // 因为下表i以后的数值都向前移动了一位，所以i也向前移动一位
                size--; // 此时数组的大小-1
            }
        }
        return size;

    }
};

题目2 ：

给定一个数组 nums，编写一个函数将所有 0 移动到数组的末尾，同时保持非零元素的相对顺序。

示例:

输入: [0,1,0,3,12]
输出: [1,3,12,0,0]
说明:

必须在原数组上操作，不能拷贝额外的数组。
尽量减少操作次数。(来源：力扣（LeetCode）)

解法一：双向箭头，向中间走

class Solution {
public:
    void moveZeroes(vector<int>& nums) 
    {
        int end = nums.size()-1;
        int beg = 0;

        if(nums.size()>=2)
        { 
            //双向指针，判断队头
            while(beg<end){
                if(nums[beg] == 0){
                    nums.push_back(0);
                    nums.erase(nums.begin()+beg);
                    end--;
                }
                //队头的元素为0 ，即可不变，反之加一
                else{
                    beg ++;
                }
                //双尾指针，判断队尾
                if(nums[end] == 0&&beg!=end){
                    nums.push_back(0);
                    nums.erase(nums.begin()+end);
                    end--;
                } 
          
            }

        }

        
    }
};

解法二：快慢指针

class Solution {
public:
    void moveZeroes(vector<int>& nums) 
    {
        int slow_index = 0;
        for(int faster_index =0;faster_index<nums.size();faster_index++){
            if(nums[faster_index]!=0){
                nums[slow_index++] = nums[faster_index];
            }
        }
        for(;slow_index<nums.size();slow_index++){
            nums[slow_index]=0;
        }
 
    }
};