来自网上,据说是Amazon的面试题。数组有N+M个数字,数字的范围为1 ... N, 打印重复的元素,要求O(M + N),O(1)空间。
像这样处理数字,要求O(n)时间复杂度,O(1)空间复杂度,基本上就是计数排序的思路吧。每个数字都入位之后,后面的就是重复元素。这里假设数组包含从1到N的所有数字。
void duplicateNumbers(int *a, int size, int max_val) {
if (a == NULL || size <= 0 || size < max_val) {
return;
}
for (int i = 0; i < size; ++i) {
while (a[i] != i+1 && a[i] != a[a[i]-1]) {
swapInArray(a, i, a[i]-1);
}
}
cout << "The dupilicate values in array are: ";
for (int i = max_val; i < size; ++i) {
cout << a[i] << " ";
}
cout << endl;
}
void duplicateNumbersTest() {
int a[] = {5, 1, 2, 2, 3, 4, 3, 6, 7, 6};
duplicateNumbers(a, sizeof(a)/sizeof(int), 7);
}