移除数组中与给定值相同的元素[LeetCode]

203 阅读1分钟

题目描述

思路

遍历整个数组,如果该值与给定值一致,则忽略,继续遍历。
如果不一致,那么该值保留下来,保留的方法是用一个“指针”i,设定初始值为0,遇到需要保留的就把nums[i]赋为遍历到的值,然后让i自增,继续找下一个需要保留的值,再赋给nums[i],这样,相当于把所有与给定值不一致的值从头到尾重新放入nums数组里了。最终的数组就是原数组中,从0到索引下表为i-1的这一段。
总结就是:

代码实现

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int i = 0;
        for (int j = 0; j < nums.size(); j++) {
            if (nums[j] != val) {
                nums[i] = nums[j]; // 关键就是这一步将值放入一个递增的i中
                i++;
            }
        }
        return i;
    }
};