CF501D Misha and Permutations Summation(康托展开)

将一个排列映射到一个数的方法就叫做康托展开。它的具体做法是这样的,对于一个给定的排列{ai}(i=1,2,3...n),对于每个ai求有多少个aj,使得j>i且ai>aj,简单来说就是求ai后面有多少数比ai小,假设我们求出来了这样的排列的一个对应数组{bi},其中bi就是ai后面有多少个数比它小。那么这个排列对应的康托展开即为∑bi*(n-i)!.

ai={1 3 5 4 2}

bi={0 1 2 1 0} 对应的排列数  0*4!+1*3!+2*2!+1*1!+0*0!.

bi可以通过树状数组在O(nlogn)里得到。

至于逆康托展开,就是已知bi,反求出ai,即知道了{0 1 2 1 0}.

做法也是类似的,我们看第一个数是0,表示后面有0个比它小的数,那么它必然是1.然后看第二个数是1,表示这个数在除去1 后面有1个比它小的数,那么这个数就是3,然后2,表示这个数后面在除去1,3,之后有2个比它小的数,所以它是5,以此往下。

一个O(nlogn)的想法是这样的,建立一棵平衡树,把1~n都插进去,每次查找第(bi+1)大的数,即为第i个数,然后把这个数删掉,以此往复。

但手写一个支持找第k大的平衡树比较慢,另外一个想法是在bit里把每个数都置1,然后二分出第一个使得前缀和为bi+1的数,这样做的复杂度就变成了O(nlog^2n).

#pragma warning(disable:4996)

#include <iostream>

#include <cstring>

#include <string>

#include <vector>

#include <cstdio>

#include <cmath>

#include <algorithm>

using namespace std;



#define maxn 210000



int bit[maxn];

int n;



void upd(int i)

{

	while (i <= n){

		bit[i] += 1;

		i += i&(-i);

	}

}



void dec(int i)

{

	while (i <= n){

		bit[i] -= 1;

		i += i&(-i);

	}

}



int cal(int i)

{

	int ret = 0;

	while (i > 0){

		ret += bit[i];

		i -= i&(-i);

	}

	return ret;

}



int solve(int x)

{

	int l = 1, r = n+1;

	while (l < r)

	{

		int mid = (l + r) >> 1;

		int tx = cal(mid);

		if (tx < x) l = mid + 1;

		else r = mid;

	}

	return l;

}



int a[maxn];

int b[maxn];

int c[maxn];

int va[maxn];

int vb[maxn];

int vc[maxn];



int main()

{

	while (cin >> n)

	{

		for (int i = 0; i < n; ++i){

			scanf("%d", a + i); a[i]++;

		}

		for (int i = 0; i < n; ++i){

			scanf("%d", b + i); b[i]++;

		}

		memset(bit, 0, sizeof(bit));

		for (int i = n - 1; i >= 0; --i){

			va[i] = cal(a[i]);

			upd(a[i]);

		}

		memset(bit, 0, sizeof(bit));

		for (int i = n - 1; i >= 0; --i){

			vb[i] = cal(b[i]);

			upd(b[i]);

		}

		reverse(va, va + n);

		reverse(vb, vb + n);

		memset(vc, 0, sizeof(vc));

		int carry = 0;

		for (int i = 1; i < n; ++i){

			vc[i] = (carry + va[i] + vb[i]) % (i + 1);

			carry = (carry + va[i] + vb[i]) / (i + 1);

		}

		reverse(vc, vc + n);

		memset(bit, 0, sizeof(bit));

		for (int i = 1; i <= n; ++i){

			upd(i);

		}

		for (int i = 0; i < n; ++i){

			c[i] = solve(vc[i]+1);

			dec(c[i]);

		}

		for (int i = 0; i < n; ++i){

			if (i) printf(" ");

			printf("%d", c[i]-1);

		}

		puts("");

	}

	return 0;

}

 

你可能感兴趣的:(mat)