解題思路
雙指針法
i爲慢指針,j爲快指針
當nums[j]==nums[i-1]時,直接跳過,如果不相等,則複製nums[j]給nums[++i],並同時遞增兩個索引,重複這一過程,直到j到達數組的末尾,該數組的新長度爲i
對於上述的具體解釋可參考:雙指針圖解
由此題的分析我們可以知道,如果要使得每個元素最多出現n次,則將nums[j]和nums[i-n+1]進行比較
代碼
class Solution {
public:
int removeDuplicates(vector<int>& nums) {
if (nums.size() <= 2) return nums.size();
int i = 1;//慢指針
for (int j=2;j<nums.size();j++)//快指針
{
if (nums[j]!=nums[i-1])//和慢指針所指的前一個元素不同
{
nums[++i] = nums[j];
}
}
return i+1;
}
};