[bzoj3439][Tire][主席树]Kpm的MC密码

Description

背景

想Kpm当年为了防止别人随便进入他的MC，给他的PC设了各种奇怪的密码和验证问题（不要问我他是怎么设的。。。），于是乎，他现在理所当然地忘记了密码，只能来解答那些神奇的身份验证问题了。。。

描述

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啦~~）

Input

第一行一个整数 n 表示字符串的数目

接下来第二行到n+1行总共n行，每行包括一个字符串，第i+1行的字符串表示编号为i的字符串

接下来包括n行，每行包括一个整数ki，意义如上题所示

Output

包括n行，第i行包括一个整数，表示所有是si的kpm串的字符串的编号中第ki小的数

Sample Input

3

cd

abcd

bcd

2

3

1

Sample Output

2

-1

2

HINT

样例解释

“cd”的kpm 串有”cd”,”abcd”,”bcd”,编号为1,2,3，第2小的编号是

2，”abcd”的kpm串只有一个，所以第3小的编号不存在，”bcd”的kpm

串有”abcd”,”bcd”,第1小的编号就是2。

数据范围与约定

设所有字符串的总长度为len,len<=300000

对于100%的数据，1<=n<=100000,0

题解

把字符串反转然后就变成了前缀问题了
对这些字符串建Tire，结束节点打标记记录一下编号。如果有相同的子串的话开一个pre[i]表示与编号i字符串上一个相同的字符串的编号是多少
由于之前看错题了我以为自己是别人的后缀那么我就是他的kpm串
然后样例秒wa发现原来是别人是我的kpm串
那么对Tire进行dfs日常主席树上树，建好之后把答案弄出来再回溯即可

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
using namespace std;
int pre[310000],n;
struct Tire
{
    int son[30],s;
    Tire(){memset(son,-1,sizeof(son));s=0;}
}tr[310000];int tot,root;
char ch[310000];
void add(int op)
{
    int len=strlen(ch+1),p=root;
    for(int i=1;i<=len;i++)
    {
        int y=ch[i]-'a'+1;
        if(tr[p].son[y]==-1)tr[p].son[y]=++tot;
        p=tr[p].son[y];
    }
    if(tr[p].s!=0)pre[op]=tr[p].s;
    tr[p].s=op;
}
struct node
{
    int lc,rc,c;
}a[8110000];int trlen;
int rt[310000];
void ins(int &now,int l,int r,int p)
{
    if(now==0)now=++trlen;
    a[now].c++;
    if(l==r)return ;
    int mid=(l+r)/2;
    if(p<=mid)ins(a[now].lc,l,mid,p);
    else ins(a[now].rc,mid+1,r,p);
}
void merge(int &x,int y)
{
    if(x==0){x=y;return ;}
    if(y==0)return ;
    a[x].c+=a[y].c;
    merge(a[x].lc,a[y].lc);
    merge(a[x].rc,a[y].rc);
}
int findKth(int x,int l,int r,int k)
{
    if(k>a[x].c)return -1;
    if(l==r)return l;
    int lc=a[x].lc,rc=a[x].rc;
    int mid=(l+r)/2;
    if(k<=a[lc].c)return findKth(lc,l,mid,k);
    else return findKth(rc,mid+1,r,k-a[lc].c);
}
int P[310000],answer[310000];
void dfs(int x)
{
    int u=tr[x].s;
    if(u!=0)ins(rt[x],1,n,u);
    while(pre[u]!=-1)ins(rt[x],1,n,pre[u]),u=pre[u];
    for(int i=1;i<=26;i++)if(tr[x].son[i]!=-1)dfs(tr[x].son[i]),merge(rt[x],rt[tr[x].son[i]]);
    u=tr[x].s;
    if(u!=0)
    {
        answer[u]=findKth(rt[x],1,n,P[u]);
        while(pre[u]!=-1)answer[pre[u]]=findKth(rt[x],1,n,P[pre[u]]),u=pre[u];
    }
}
int main()
{
    scanf("%d",&n);root=0;
    memset(pre,-1,sizeof(pre));
    for(int i=1;i<=n;i++)
    {
        scanf("%s",ch+1);int len=strlen(ch+1);
        for(int j=1;j<=len/2;j++)swap(ch[j],ch[len-j+1]);
        add(i);
    }
    for(int i=1;i<=n;i++)scanf("%d",&P[i]);
    dfs(root);
    for(int i=1;i<=n;i++)printf("%d\n",answer[i]);
    return 0;
}