No.27 Remove Element

No.27 Remove Element

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.

Tags： Array Two Pointers

移除数组中所有的给定数字
要求：原地移除，不能使用额外空间；返回新数组的长度;可以改变原始顺序

典型的两指针

 

#include "stdafx.h"
#include <map>
#include <vector>
#include <iostream>
using namespace std;

class Solution
{
public:
    int removeElement(vector<int> &nums, int val)
    {//移除数组中所有的给定数字
     //要求：原地移除，不能使用额外空间；返回新数组的长度;可以改变原始顺序
        int size = nums.size();
        int count = 0;//计数当前val出现的次数

        for(int i=0; i<size; i++)
        {
            if(nums[i] == val)
                count++;
            else
            {
                if(count != 0)
                    nums[i-count] = nums[i];
            }
        }
        nums.erase(nums.begin()+size-count,nums.end());
        return size-count;
    }
};

int main()
{
    Solution sol;
    int data[] = {1,2,2,3,3,2,1,7,9,10};
    vector<int> test(data,data+sizeof(data)/sizeof(int));
    for(const auto &i : test)
        cout << i << " ";
    cout << endl;
    cout<< boolalpha << sol.removeElement(test,2)<<endl;
        for(const auto &i : test)
        cout << i << " ";
    cout << endl;    
}