Careercup - Facebook面试题 - 4909367207919616

2014-05-01 01:23

题目链接

原题：

WAP to modify the array such that arr[I] = arr[arr[I]].
Do this in place i.e. with out using additional memory.

example : if a = {2,3,1,0}
o/p = a = {1,0,3,2}

Note : The array contains 0 to n-1 integers.

题目：给定一个长度为n的数组，由0至n - 1的排列组成。请做这样的变换，arr[i] = arr[arr[i]]。要求就地完成，不用额外数组。

解法：当数据范围有明确限制时，能够通过压缩维度来节省空间。对于一个数对(x, y)，如果x和y都属于[0, n - 1]，那么x * n + y就能唯一表示这个数对了。因此，可以先把变换后的结果存在高位上，然后再把高位的数通过除法移动到低位。算法是线性的，能够就地完成。

代码：

// http://www.careercup.com/question?id=4909367207919616 #include <cstdio>
#include <vector>
using namespace std;

void displaceInPlace(vector<int> &v)
{
int i;
int n;

// all elements in the array has value between 0 and n - 1.
n = (int)v.size();
for (i = 0; i < n; ++i) {
v[i] = v[v[i]] % n * n + v[i];
}
for (i = 0; i < n; ++i) {
v[i] = v[i] / n;
}
}

int main()
{
int i, n;
vector<int> v;

while (scanf("%d", &n) ==  1 && n > 0) {
v.resize(n);
for (i = 0; i < n; ++i) {
scanf("%d", &v[i]);
}
displaceInPlace(v);
for (i = 0; i < n; ++i) {
printf((i ? " %d" : "%d"), v[i]);
}
putchar('\n');
v.clear();
}

return 0;
}