题目是Follow up for "Remove Duplicates":
What if duplicates are allowed at most twice?
For example,
Given sorted array A =
[1,1,1,2,2,3]
,
Your function should return length =
5
, and A is now [1,1,2,2,3]
.
费了好长好长时间,脑子总是转不过来。最后终于写出来了。增加了2个变量,一个计数器,一个index标明存放下个不同数据的位置。
如果是没有排序的数组,则需要引入一个hashmap 来记录出现次数。
public class Solution {
public int removeDuplicates(int[] A) {
int index=1;
int samenumber=0;
if(A.length==0) return 0;
for(int i=1;i<A.length;i++){
if(A[i]==A[i-1]) {
samenumber++;
if(samenumber==1) {
A[index] = A[i];
index++;
}
}
if(A[i]!=A[i-1]){
A[index]=A[i];
index++;
samenumber=0;
}
}
return index;
}
}
代码虽然长,不过被leetcode接受就可以可以了,扩展性好些。如果这题改成允许每个字符相同最多3次,程序只要改一处地方 if(samenumber==1) 改成if(samenumber<=2) :
public int removeDuplicates4(int[] A) {
int index=1;
int samenumber=0;
if(A.length==0) return 0;
for(int i=1;i<A.length;i++){
if(A[i]==A[i-1]) {
samenumber++;
if(samenumber<=2) {
A[index] = A[i];
index++;
}
}
if(A[i]!=A[i-1]){
A[index]=A[i];
index++;
samenumber=0;
}
}
return index;
}
加一个别人写的代码c++的代码
// LeetCode, Remove Duplicates from Sorted Array II
// 时间复杂度O(n),空间复杂度O(1)
// @author hex108 (https://github.com/hex108)
// 时间复杂度O(n),空间复杂度O(1)
// @author hex108 (https://github.com/hex108)
class Solution {
public:
int removeDuplicates(int A[], int n) {
if (n <= 2) return n;
int index = 2;
for (int i = 2; i < n; i++){
if (A[i] != A[index - 2])
A[index++] = A[i];
}
return index;
}
};