#include<iostream>
#include<vector>
using namespace std;
class Solution {
public:
int removeDuplicates(vector<int>& nums)
{
int num = nums.size();//计算删除重复元素数组中的元素个数
int j = 0;
for (int i = 0; i < nums.size()-1; i++)
{
j = i + 1;
while (nums[i] == nums[j])
{
for (; j+1< nums.size(); j++)
{
nums[j] = nums[j + 1];
}
vector<int>::iterator it = nums.begin() + j;
nums.erase(it);
num--;
j=i+1;
}
}
return num;
}
};
void test()
{
vector<int> arr = { 1,1,2 };
Solution s;
int num=s.removeDuplicates(arr);
cout << "数组元素个数为:" << num << endl;
for (int i = 0; i < num; i++)
{
cout << arr[i];
}
cout << endl;
}
int main()
{
test();
system("pause");
return 0;
}
双指针法
首先注意数组是有序的,那么重复的元素一定会相邻。
要求删除重复元素,实际上就是将不重复的元素移到数组的左侧,即慢指针p的右边都是不重复的元素,p—q之间是出现重复的元素。
考虑用 2 个指针,一个在前记作 p,一个在后记作 q,算法流程如下:
1.比较 p 和 q 位置的元素是否相等。
如果相等,q 后移 1 位
如果不相等,将 q 位置的元素复制到 p+1 位置上,p 后移一位,q 后移 1 位
重复上述过程,直到 q 等于数组长度。
返回 p + 1,即为新数组长度。
int removeDuplicates(vector<int>&nums) {
if (nums.size() < 2) return nums.size();
int j = 0;
for (int i = 1; i < nums.size(); i++)
if (nums[j] != nums[i]) nums[++j] = nums[i];
return ++j;
}