[洛谷U72177]火星人plus

题目大意:给你一个$1\sim n(n\leqslant 10^5)$的排列,设$a$为它在$1\sim n$的全排列中的排名,求在$1\sim n$的全排列中第$a+m$个排列。

题解:康托展开以及逆康托展开。将原排列转为变进制数,加上$m$,再用转回排列。转回去可以用在树状数组上二分来解决。这里使用了$skip2004$教的两种方法。

卡点:变进制数加法时写错

C++ Code:

#include <cstdio>
#include <iostream>
#include <algorithm>
const int maxn = 1 << 17;

int n, L = 1, nn;

namespace BIT {
	int V[maxn], res;
	inline void inc(int p) { for (; p <= L; p += p & -p) ++V[p]; }
	inline void dec(int p) { for (; p <= L; p += p & -p) --V[p]; }
	inline int sum(int p) { for (res = 0; p; p &= p - 1) res += V[p]; return res; }
	inline void fill(int p, int L) { for (int i = 0; i < L; ++i) V[i] = p * (i & -i); }
	inline int query(int k) { // 树状数组上二分
		// 区间法
		static int l, r, mid;
		l = 1, r = L;
		while (l != r) {
			mid = l + r >> 1;
			if (V[mid] < k) k -= V[mid], l = mid + 1;
			else r = mid;
		}
		dec(l);
		return l;
		// 跳点法
		static int rt; rt = L;
		for (int i = rt; i >>= 1; ) {
			if (V[rt - i] < k) k -= V[rt - i];
			else rt -= i;
		}
		dec(rt);
		return rt;
	}
}

long long a[maxn], m;
int main() {
	std::ios::sync_with_stdio(false), std::cin.tie(0), std::cout.tie(0);
	std::cin >> n >> m;
	while (L < n) L <<= 1;
	for (int i = 1, t; i <= n; ++i) {
		std::cin >> a[i];
		t = BIT::sum(a[i]);
		BIT::inc(a[i]);
		a[i] = a[i] - t - 1;
	}
	a[n] += m;
	for (int i = n; i; --i) {
		a[i - 1] += a[i] / (n - i + 1);
		a[i] %= n - i + 1;
	}
	BIT::fill(1, L + 1);
	for (int i = 1; i <= n; ++i) std::cout << BIT::query(a[i] + 1) << ' ';
	std::cout << '\n';
	return 0;
}

  

猜你喜欢

转载自www.cnblogs.com/Memory-of-winter/p/11164672.html