【移除元素】-CSDN博客

本文链接：https://blog.csdn.net/qq_50728732/article/details/138959745

27.移除元素

题目描述

给你一个数组 nums 和一个值 val，你需要原地移除所有数值等于 val 的元素。元素的顺序可能发生改变。然后返回 nums 中与 val 不同的元素的数量。

假设 nums 中不等于 val 的元素数量为 k，要通过此题，您需要执行以下操作：

更改 nums 数组，使 nums 的前 k 个元素包含不等于 val 的元素。nums 的其余元素和 nums 的大小并不重要。
返回 k。

思路

暴力解法
用两个循环，外层循环遍历数组，内层循环移动元素。请添加图片描述

错误

27.超时错误

public static int removeElement(int[] nums, int val) {
        int count = 0;
        for (int i = 0;i<nums.length;i++){
            if(nums[i] == val){
                for (int j = i+1;j<nums.length;j++){
                    nums[j-1] = nums[j];
                }
                count++;
                i--;// 因为下标i以后的数值都向前移动了一位，所以i也向前移动一位
            }
        }
        return nums.length-count;
    }

起初我把自己的代码跟题解反复对比都没有找出错误，一点点改，发现陷入死循环的原因处在nums.length

  for (int j = i+1;j<nums.length;j++)

题解中定义变量size为nums.length，后续遇到nums[i]==val时，size--，这里并不能完全用count++替代，因为size影响后面的第二个循环。

// 时间复杂度：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;

    }
};