[SOJ #687]双生串(2019-11-6考试)/[hdu5431]AB String

题目大意

把所有仅包含(AB)的字符串按字典序排列，给你一个仅包含(AB)的字符串(S)，然后有(Q)个问题，第(i)个问题给你(k_i)，求不是(S)的子串中，第(k_i)小的是什么。(T)组数据

(Tleqslant5)，(sum|S_i|leqslant2.3 imes10^5)，(Q_ileqslant10)，(k_ileqslant10^9)

题解

发现，长度为(l)的字符串有(2^l)个，而(S)的长度小于等于(l)的子串最多有(l|S|)个，可以发现，当(l>31)时(2^l-l|S|>10^9)，即答案长度一定小于等于(31)。因为字符串中只含有(AB)，可以用二进制表示，用(mathrm{hash})算出(S)中每种长度的子串。这样可以算出最终的答案的长度，然后在这一个长度中二分即可。

卡点

多测一定要清空！！

C++ Code：

#include <cstdio>
#include <iostream>
#include <algorithm>

int T, n, Q, k, p;
int s[32][250000], cnt[32];
std::string __s;
bool check(int mid, int p) {
	return mid + 1 -
		(std::upper_bound(s[p], s[p] + cnt[p], mid) - s[p]) >= k;
}
int main() {
	std::ios::sync_with_stdio(false), std::cin.tie(0), std::cout.tie(0);
	std::cin >> T;
	while (T --> 0) {
		std::cin >> __s, n = __s.length();
		for (int i = 1; i <= 31 && i <= n; ++i) {
			const int I = (1 << i) - 1; s[i][0] = 0;
			for (int j = 0; j < i; ++j) s[i][0] = s[i][0] << 1 | __s[j] - 'A';
			for (int j = i; j < n; ++j)
				s[i][j - i + 1] = (s[i][j - i] << 1 | __s[j] - 'A') & I;
			std::sort(s[i], s[i] + n - i + 1);
			cnt[i] = std::unique(s[i], s[i] + n - i + 1) - s[i];
		}
		std::cin >> Q;
		while (Q --> 0) {
			std::cin >> k;
			for (p = 1; p <= 31; ++p)
				if ((1 << p) - cnt[p] < k) k -= (1 << p) - cnt[p];
				else break;
			int l = 0, r = (1 << p) - 1, ans = -1;
			while (l <= r) {
				int mid = l + r >> 1;
				if (check(mid, p)) ans = mid, r = mid - 1;
				else l = mid + 1;
			}
			for (int i = p - 1; ~i; --i) std::cout << "AB"[ans >> i & 1];
			std::cout.put('
');
		}
		for (int i = 1; i <= 31; ++i) cnt[i] = 0;
	}
	return 0;
}