当前位置：首页 > article >正文

27. 移除元素

article 2024/9/20 23:38:09

我的思路：
很绕，且有漏洞
使用t记录第一个需要覆盖的元素的位置，cnt表示待覆盖的元素的数量，但是在[4,5] val=4中，当i遍历到5时，cnt仍然等于1，于是t又变成了i的值，指向1，于是出错。

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int len = nums.size();
        sort(nums.begin(), nums.end());
        int cnt = 0, t = -1;
        for (int i = 0; i < len; ++i) {
            if (nums[i] == val) ++cnt;
            if (cnt == 1) {
                t = i;
            }
            if (nums[i] != val && t >= 0) {
                nums[t] = nums[i];
                ++t;
            } 
        }
        return len - cnt;
    }
};

正常可以直接使用cnt作为待覆盖的元素的下标，将所有不等于val的元素覆盖到cnt的位置。

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