算法学习-Cantor数组

题目

已知数组A[0...N-1]乱序着前N个正整数，现统计后缀数组A[i+1...N-1]中小于元素A[i]的数目，并存放在数组C[i]中。如给定数组A={4,6,2,5,3,1},得到数组C={3,4,1,2,1,0}。

问：给定数组C={3,4,1,2,1,0},如何恢复数组A？

我们称A为原数组，C为Cantor数组。

求Cantor数组代码如下

void CantorExpansion(const int* a, int* b, int size)
{
	int i, j;
	for (i = 0; i < size; i++)
	{
		b[i] = 0;
		for (j = i + 1; j < size; j ++)
		{
			if (a[j] < a[i])
				b[i]++;
		}
	}
}

问题分析

Cantor数组：{3,4,1,2,1,0}

原数组：{4,6,2,5,3,1}

给定顺序数组B={1,2,3...N-1,N},从0开始数

考察Cantor数组的首位C[0]:

小于A[0]的个数为C[0]，则A[0]为B[C[0]]

在序列数组B中删除B[C[0]],仍然满足以上性质。

代码

void CantorExpansionR(const int* a, int* result, int size)
{
	int i;
	vector<int> v(size);
	for (i = 0; i < size; i++)
		v[i] = i + 1;
	for (i = 0; i < size; i++)
	{
		result[i] = v[a[i]];
		v.erase(v.begin() + a[i]);
	}
}

进一步思考

以上代码空间复杂度为O（N），时间复杂度为O（N2），若允许更改数组C，可否降低空间复杂度？

考察Cantor数组中第一个出现0的位置：他表示位于该位置右侧的所有元素都大于该元素，则该元素必然是最小的。每次找到一个0后，将0左侧的Cantor值都减一，重复以上操作。

空间复杂度为O（1）。

代码如下

void CantorExpansionR2(int* a, int* result, int size)
{
	memset(result, 0, sizeof(int)*size);
	int i, j;
	for (i = 0; i < size; i++)
	{
		for (j = 0; j < size; j++)
		{
			if (result[j] != 0)
				continue;
			if (a[j] == 0)
				break;
			a[j]--;
		}
		result[j] = i + 1;
	}
}