【leetcode】27-移除元素【C++】

题目如下:

【leetcode】27-移除元素【C++】_第1张图片

【leetcode】27-移除元素【C++】_第2张图片

解题思路:

先对数组排序,然后检索数组中瞒住条件的元素,用 count 记录满足条件的元素个数,用 flag 记录第一个满足条件元素的前一个元素的位置,之后向前移动后续的数组元素。

代码如下:

class Solution {
public:
    int removeElement(vector& nums, int val) {
        //数组为空,直接返回
        if(nums.empty()) return 0;
        int l = nums.size();
        int flag = -1, count = 0, res = 0; //当第一个元素就开始相等时,flag为-1
        //排序数组
        sort(nums.begin(), nums.end());
        //count记录满足条件的数组元素个数,flag记录第一个满足条件元素的前一个位置
        for(int i = 0; i < l; ) {
            if(nums[i] != val && count == 0){ //还未遇见相等元素
                flag = i;
                i++;
            }
            else if(nums[i] == val){ //遍历相等元素
                count++;
                i++;
            }
            else break; //相等元素全部遍历,直接退出循环
        }
        //向前移动数组元素
        for(int i = 1; flag+i+count < l; i++) {
            nums[flag + i] = nums[flag + i + count];
            res++;
        }
        return (res + flag + 1);
    }
};

【leetcode】27-移除元素【C++】_第3张图片

 

你可能感兴趣的:(leetcode算法题库)