原地数组求解例题汇总（C++力扣）

最新推荐文章于 2024-03-30 23:05:17 发布

原创最新推荐文章于 2024-03-30 23:05:17 发布

· 264 阅读

0 ·

版权

LeetCode(C++) 专栏收录该内容

64 篇文章

订阅专栏

283.移出0

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

示例:

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

必须在原数组上操作，不能拷贝额外的数组。
尽量减少操作次数。
首先确立思想，既然不能使用额外空间，那就使用两个指针。关注点在,不为零的情况下的操作，一旦不为0，就把该元素赋给以k为指针的下标数组中。最后把多余的空间单独放成0。

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

优化：
直接让他们交换，这样就不需要单独给剩下的空间赋值了。

class Solution {
public:
    void moveZeroes(vector<int>& nums) {
       int k=0;
       for(int i=0; i<nums.size(); i++){
           if(nums[i])
               swap(nums[k++],nums[i]);
       }
    }
};

再优化，如果k现在就在i所在的地方，那就不需要赋值了，直接开始下一个，让k++。

class Solution {
public:
    void moveZeroes(vector<int>& nums) {
       int k=0;
       for(int i=0; i<nums.size(); i++){
           if(nums[i])
              if(k!=i)
               swap(nums[k++],nums[i]);
               else
               k++;          
            
       }
    }
};

下面的例题都是基于上面的思想

27.移除元素

给你一个数组 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。

注意这五个元素可为任意顺序。

你不需要考虑数组中超出新长度后面的元素。

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

26.删除排序数组中的重复项

给定一个排序数组，你需要在原地删除重复出现的元素，使得每个元素只出现一次，返回移除后数组的新长度。

不要使用额外的数组空间，你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。

示例 1:

给定数组 nums = [1,1,2],

函数应该返回新的长度 2, 并且原数组 nums 的前两个元素被修改为 1, 2。

你不需要考虑数组中超出新长度后面的元素。
示例 2:

给定 nums = [0,0,1,1,1,2,2,3,3,4],

函数应该返回新的长度 5, 并且原数组 nums 的前五个元素被修改为 0, 1, 2, 3, 4。

你不需要考虑数组中超出新长度后面的元素。

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        int n=nums.size();
        int k=0;
        if(n==0)return 0;
        for(int i=0;i<n-1;i++){
            if(nums[i]!=nums[i+1]){
                if(i!=k)
               swap(nums[k++],nums[i]);
               else
               k++;
            }
        }
        nums[k]=nums[n-1];
    return k+1;
    }
};

28.删除数组中的重复项

给定一个增序排列数组 nums ，你需要在原地删除重复出现的元素，使得每个元素最多出现两次，返回移除后数组的新长度。

不要使用额外的数组空间，你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。

说明：

为什么返回数值是整数，但输出的答案是数组呢？

请注意，输入数组是以“引用”方式传递的，这意味着在函数里修改输入数组对于调用者是可见的。

你可以想象内部操作如下：

// nums 是以“引用”方式传递的。也就是说，不对实参做任何拷贝
int len = removeDuplicates(nums);

// 在函数里修改输入数组对于调用者是可见的。
// 根据你的函数返回的长度, 它会打印出数组中该长度范围内的所有元素。
for (int i = 0; i < len; i++) {
print(nums[i]);
}

示例 1：

输入：nums = [1,1,1,2,2,3]
输出：5, nums = [1,1,2,2,3]
解释：函数应返回新长度 length = 5, 并且原数组的前五个元素被修改为 1, 1, 2, 2, 3 。你不需要考虑数组中超出新长度后面的元素。
示例 2：

输入：nums = [0,0,1,1,1,1,2,3,3]
输出：7, nums = [0,0,1,1,2,3,3]
解释：函数应返回新长度 length = 7, 并且原数组的前五个元素被修改为 0, 0, 1, 1, 2, 3, 3 。你不需要考虑数组中超出新长度后面的元素。

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        int n=nums.size();
        int k=0;
        if(n<=1)return n;
        for(int i=0;i<n-2;i++){
            if(nums[i]!=nums[i+2])
            if(i!=k)
            swap(nums[k++],nums[i]);
            else
            k++;
        }
        nums[k]=nums[n-2];
        nums[k+1]=nums[n-1];
        return k+2;
    }
};