题目:
Given a sorted array, remove the duplicates in place such that each element appear only once and return the new length.
Do not allocate extra space for another array, you must do this in place with constant memory.
For example,
Given input array A = [1,1,2]
,
Your function should return length = 2
, and A is now [1,2]
.
一开始我还以为只用统计下不重复的元素个数就好了,结果发现还要对原来的数组进行移位,使得前length个元素中没有重复。- -!
如果每遇到一个重复元素,就把后面所有的元素向前移动一位,这样做的开销太大,时间复杂度为O(n2);
其实,我们只需设置两个指针i,j,一个i用来指示当前遍历到的元素的位置,一个j用来指示当前的元素真正应该放在哪个位置,只有在遍历到非重复元素时,j才会++。所以,可以知道,j永远不会超过i
public class No25_RemoveDuplicatesFromSortedArray {
public static void main(String[] args){
System.out.print(removeDuplicates(new int[]{1,2,2}));
}
public static int removeDuplicates(int[] A) {
if(A.length <= 0) return 0;
int i = 1;
int j = 1;
int pre = A[0];
for(;i<A.length;i++){
if(A[i] == pre){
continue;
}
pre = A[i];
A[j++] = A[i];
}
return j;
}
// TLE!!!
// public static int removeDuplicates(int[] A) {
// if(A.length <= 0) return 0;
// int pre = A[0];
// int ret = 1;
// for(int i=1;i<A.length;i++){
// if(A[i] == pre){
// for(int j=i+1;j<A.length;j++){
// A[j-1] = A[j];
// }
// }
// ret++;
// pre = A[i];
// }
// return ret;
// }
}