【Leetcode】之Remove Element

最新推荐文章于 2018-10-11 23:44:42 发布

原创最新推荐文章于 2018-10-11 23:44:42 发布 · 372 阅读

0 ·

CC 4.0 BY-SA版权

Leetcode 同时被 2 个专栏收录

75 篇文章

订阅专栏

算法练习

72 篇文章

订阅专栏

本文介绍了一种在数组中移除特定值并返回新长度的算法实现。该算法通过记录非目标值的位置来高效地完成任务，同时允许元素顺序发生变化。

部署运行你感兴趣的模型镜像

一.问题描述

Given an array and a value, remove all instances of that value in place and return the new length.

The order of elements can be changed. It doesn’t matter what you leave beyond the new length.

二.我的解题思路

这道题目有点像上一道题，都比较简单。我的做法是每次都记录之前一个非value数的下标pri，当遇到新的非value数的时候，直接放在pri后面即可。测试通过的程序如下：

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


        }
        return cnt;
    }
};