【后缀自动机】 HDOJ 5431 AB String

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;

const int alpha = 2;
const int maxn = 40005;
const int maxm = 30;

struct node
{
int len;
LL cnt[maxm];
node *fa, *ch[alpha];
}pool[maxn], *last, *root;

vector<int> ans;
char s[maxn];
int a[maxn];
int tot;

node* newnode(int len)
{
node *p = &pool[tot++];
p->len = len;
p->fa = 0;
memset(p->ch, 0, sizeof p->ch);
return p;
}

void init()
{
tot = 0;
root = last = newnode(0);
}

void add(int c)
{
node *p = last, *np = newnode(p->len + 1);
last = np;
for(; p && !p->ch[c]; p = p->fa) p->ch[c] = np;
if(!p) np->fa = root;
else {
node *q = p->ch[c];
if(q->len == p->len + 1) np->fa = q;
else {
node *nq = newnode(p->len + 1);
*nq = *q;
nq->len = p->len + 1;
np->fa = q->fa = nq;
for(; p && p->ch[c] == q; p = p->fa) p->ch[c] = nq;
}
}
}

int C[maxn];
node *tp[maxn];
LL two[100];
void Tp()
{
memset(C, 0, sizeof C);
for(int i = 0; i < tot; i++) C[pool[i].len]++;
for(int i = 1; i < tot; i++) C[i] += C[i-1];
for(int i = 0; i < tot; i++) tp[--C[pool[i].len]] = &pool[i];
}

void solve(int& K, int cnt)
{
K--;
for(int i = 1; i <= cnt; i++) {
a[i] = K % 2;
K /= 2;
}
for(int i = cnt; i >= 1; i--) ans.push_back(a[i] + 'A');
}

void dfs(node *o, int& K, int pos)
{
if(K <= 0) return;
if(pos == 0) return;
for(int i = 0; i < alpha; i++) {
if(o->ch[i]) {
if(K <= o->ch[i]->cnt[pos-1]) {
ans.push_back(i + 'A');
dfs(o->ch[i], K, pos-1);
return;
}
else K -= o->ch[i]->cnt[pos-1];
}
else {
if(K <= two[pos-1]) {
ans.push_back(i + 'A');
if(pos - 1) solve(K, pos - 1);
return;
}
else K -= two[pos-1];
}
}
}

void work()
{
scanf("%s", s);
for(int i = 0; s[i]; i++) add(s[i] - 'A');
Tp();
two[0] = 1;
for(int i = 1; i <= 40; i++) two[i] = (1LL << i);
for(int i = tot - 1; i >= 0; i--) {
tp[i]->cnt[0] = 0;
for(int j = 1; j < maxm; j++) {
tp[i]->cnt[j] = 0;
for(int k = 0; k < alpha; k++) {
if(tp[i]->ch[k]) tp[i]->cnt[j] += tp[i]->ch[k]->cnt[j-1];
else tp[i]->cnt[j] += two[j - 1];
}
}
}
int m, K;
scanf("%d", &m);
while(m--) {
scanf("%d", &K);
int tlen;
for(tlen = 1;; tlen++) {
if(root->cnt[tlen] < K) K -= root->cnt[tlen];
else break;
}
ans.clear();
dfs(root, K, tlen);
for(int i = 0; i < ans.size(); i++) printf("%c", ans[i]);
printf("\n");
}
}

int main()
{
int _;
scanf("%d", &_);
while(_--) {
init();
work();
}

return 0;
}