LeetCode: Remove Duplicates from Sorted Array

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].


解題思路:

移除數組中的的重複的元素,只保留相應的一個,返回一個去重後的數組。分別使用兩個索引,一個用於遍歷數組,一個用於保存不重複數組的下標,即可在O(n)的時間複雜度內完成去重操作,空間複雜度爲O(1)


代碼如下:

int removeDuplicates(int A[], int n) {
	if (n == 0) {
		return 0;
	}
	int index = 0;
	for (int i=1; i<n; i++) {
		if (A[index] != A[i]) {
			A[++index] = A[i];
		}
	}
	return index + 1;
}

第二種方法:

使用STL,時間複雜度也爲O(n),空間複雜度爲O(1)

int removeDuplicates(int A[], int n) {
	return distance(A, unique(A, A + n))
}



發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章