题目:
解答:
移除重复的元素 的升级版,思路和前面的题类似 用first指针指向最后一个排好序的元素,last指针指向第一个未处理的元素。
额外的,我们用dump变量来记录最后的first指针指向的元素重复了几次。
如果是新元素或者重复一次,则要添加到first后,重复两次的则继续last++;
初始化first = 0, last = 1 , dump = 1;表示已经处理完第一个元素。
代码:
class Solution {
public:
<span style="white-space:pre"> </span> int removeDuplicates(int A[], int n) {
<span style="white-space:pre"> </span> int first = 0; //初始 = 0,有序数组为1 ,已经处理完第一个元素
<span style="white-space:pre"> </span> int last = 1; //未处理数组从1开始
<span style="white-space:pre"> </span> if (n == 0)
<span style="white-space:pre"> </span> return 0;
<span style="white-space:pre"> </span> int dump = 1; //已经处理完第一个元素,初始重复数 = 1
<span style="white-space:pre"> </span> while (last < n)
<span style="white-space:pre"> </span> {
<span style="white-space:pre"> </span> if (A[first] == A[last])
<span style="white-space:pre"> </span> {<span style="white-space:pre"> </span> if (dump == 2) //如果已重复两次 则忽略该元素
<span style="white-space:pre"> </span>++last;
<span style="white-space:pre"> </span> else //移动该元素,++dump
<span style="white-space:pre"> </span> {
<span style="white-space:pre"> </span> A[first+1] = A[last++];
<span style="white-space:pre"> </span> ++first;
<span style="white-space:pre"> </span> ++dump;
<span style="white-space:pre"> </span> }
<span style="white-space:pre"> </span> }
<span style="white-space:pre"> </span> else //出现了不相同的新元素,
<span style="white-space:pre"> </span> {
<span style="white-space:pre"> </span> A[first+1] = A[last++];
<span style="white-space:pre"> </span> ++first;
<span style="white-space:pre"> </span> dump = 1;
<span style="white-space:pre"> </span> }
<span style="white-space:pre"> </span> }
<span style="white-space:pre"> </span> return first + 1;
<span style="white-space:pre"> </span> }
};