[BZOJ 3439]Kpm的MC密碼

Kpm當年設下的問題是這樣的：

    現在定義這麼一個概念，如果字符串s是字符串c的一個後綴，那麼我們稱c是s的一個kpm串。

    系統將隨機生成n個由a…z組成的字符串，由1…n編號（s1,s2…,sn），然後將它們按序告訴你，接下來會給你n個數字，分別爲k1…kn，對於每一個ki，要求你求出列出的n個字符串中所有是si的kpm串的字符串的編號中第ki小的數，如果不存在第ki小的數，則用-1代替。（比如說給出的字符串是cd,abcd,bcd,此時k1=2，那麼”cd”的kpm串有”cd”,”abcd”,”bcd”,編號分別爲1,2,3其中第2小的編號就是2）（PS：如果你能在相當快的時間裏回答完所有n個ki的查詢，那麼你就可以成功幫kpm進入MC啦~~）

我累個去你有重串不記得說一聲！！***(雖然寫的時候考慮了一下但是由於蒟蒻太懶自認爲沒有重串)

調死辣。。

呢。倒着建棵Tire

然後詢問dfs序上一段區間的最第k小值

主席樹妥妥的

#include 
#define maxn 500010
using namespace std;

int n, root, size;

int t[maxn][26], mark[maxn];
int Map[maxn];

struct Edge{
    int to, next;
}edge[maxn];
int h[maxn], cnt;
void add(int u, int v){
    cnt ++;
    edge[cnt].to = v;
    edge[cnt].next = h[u];
    h[u] = cnt;
}

char s[maxn];

int par[maxn];
int getfa(int x){return par[x] == x ? x : par[x] = getfa(par[x]);}

void Insert(int Id){
    int N = strlen(s+1), now = root;
    for(int i = N; i >= 1; i --){
        int p = s[i] - 'a';
        if(!t[now][p])t[now][p] = ++ size;
        now = t[now][p];
    }
    if(!mark[now]){
		mark[now] = true;
		Map[now] = Id;
    }
    else{par[getfa(Id)] = Map[now];}
}

int que[maxn], fa[maxn];

vectortmp[maxn];

void BFS(){
    int head = 0, tail = 0, u, v;
    que[tail ++] = root;
    while(head < tail){
        u = que[head ++];
        if(mark[u])add(Map[fa[u]], Map[u]);
        for(int i = 0; i < 26; i ++){
            if(v = t[u][i]){
                que[tail ++] = v;
                if(mark[u])fa[v] = u;
                else fa[v] = fa[u];
            }
        }
    }
    for(int i = 1; i <= n; i ++)
        if(getfa(i) != i)tmp[getfa(i)].push_back(i);
}

int In[maxn], Out[maxn], dfs_clock, dfn[maxn];

void DFS(int u){
    In[u] = ++ dfs_clock;
    dfn[dfs_clock] = u;
    for(int i = 0; i < tmp[u].size(); i ++){
		int v = tmp[u][i];
		In[v] = In[u];
		dfn[++ dfs_clock] = v;
    }
    for(int i = h[u]; i; i = edge[i].next)
        DFS(edge[i].to);
    Out[u] = dfs_clock;
    for(int i = 0; i < tmp[u].size(); i ++){
		int v = tmp[u][i];
		Out[v] = dfs_clock;
    }
}

namespace Segtree{
    #define M 3000000
    int L[M], R[M], C[M], T[maxn], size, k;
	int build(int l, int r){
		int root = ++ size;
		if(l == r)return root;
		int mid = l + r >> 1;
		L[root] = build(l, mid), R[root] = build(mid+1, r);
		return root;
	}
	
    int add(int root, int l, int r, int pos, int val){
        int cur = ++ size;
        L[cur] = L[root], R[cur] = R[root], C[cur] = C[root] + val;
        if(l == r)return cur;
        int mid = l + r >> 1;
        if(pos <= mid)L[cur] = add(L[root], l, mid, pos, val);
        else R[cur] = add(R[root], mid+1, r, pos, val);
        return cur;
    }

    int Find(int a, int b, int l, int r){
        if(l == r)return l;
        int mid = l + r >> 1;
        if(k <= C[L[b]] - C[L[a]])return Find(L[a], L[b], l, mid);
        k -= C[L[b]] - C[L[a]];return Find(R[a], R[b], mid+1, r);
    }

    void work(){
		T[0] = build(1, n);
        for(int i = 1; i <= dfs_clock; i ++)
            if(dfn[i])T[i] = add(T[i-1], 1, n, dfn[i], 1);
        for(int i = 1; i <= n; i ++){
            scanf("%d", &k);
            int l = In[i]-1, r = Out[i];
            if(C[T[r]] - C[T[l]] >= k)
				printf("%d\n", Find(T[l], T[r], 1, n));
            else printf("-1\n");
        }
    }
}

int main(){
    scanf("%d", &n);
    for(int i = 1; i <= n; i ++)
        par[i] = i;
    for(int i = 1; i <= n; i ++)
        scanf("%s", s+1), Insert(i);
    BFS();
    DFS(root);
    Segtree::work();
    return 0;
}