3439: Kpm的MC密码 trie+主席树

一开始写的fail树，然后沿着fail指针反向dfs。结果TLE啦。
看了一下题解，是把后缀相同转化为前缀相同，然后在trie树上有相同前缀的肯定在同一棵子树中，按dfs序建主席树就行啦。
其实在fail树上建主席树也是可以的。。一开始傻逼没有想到。。懒得改了。。
注意有相同的子串！

fail树暴力

#include<iostream>#include<cstdio>#include<algorithm>#include<vector>#include<cstring>#define N 100005using namespace std;int n,cnt,top;int a[N][26],p[N],pos[N],q[N],stack[N];bool flag[N];char s[N];vector<int> b[N];vector<int> id[N];inline int read(){    int a=0,f=1; char c=getchar();    while (c<'0'||c>'9') {if (c=='-') f=-1; c=getchar();}    while (c>='0'&&c<='9') {a=a*10+c-'0'; c=getchar();}    return a*f;}inline void insert(int x){    scanf("%s",s);    int c,now=1,len=strlen(s);    for (int i=0;i<len;i++)    {        c=s[i]-'a';        if (a[now][c]) now=a[now][c];        else now=a[now][c]=++cnt;    }    id[now].push_back(x); pos[x]=now; flag[now]=1;}inline void build_fail(){    int t=0,w=1,now;    q[1]=1; p[1]=0;    while (t<w)    {        now=q[++t];        for (int i=0;i<26;i++)            if (a[now][i])            {                int k=p[now];                while (!a[k][i]) k=p[k];                p[a[now][i]]=a[k][i];                if (a[k][i]!=1) b[a[k][i]].push_back(a[now][i]);                q[++w]=a[now][i];            }    }}void dfs(int x){    if (flag[x])         for (int i=0;i<id[x].size();i++) stack[++top]=id[x][i];    for (int i=0;i<b[x].size();i++) dfs(b[x][i]);}int main(){//  freopen("password.in","r",stdin);//  freopen("password.out","w",stdout);    n=read(); cnt=1;    for (int i=0;i<26;i++) a[0][i]=1;    for (int i=1;i<=n;i++) insert(i);    build_fail();    for (int i=1;i<=n;i++)     {        top=0;        dfs(pos[i]);        sort(stack+1,stack+top+1);        int k=read();        if (k>top) puts("-1");        else printf("%d\n",stack[k]);    }    return 0;}

#include<iostream>#include<cstdio>#include<algorithm>#include<vector>#include<cstring>#define N 100005using namespace std;int n,cnt,dfn,size;int a[N][26];int in[N],out[N],root[N],pos[N],p[N];char s[N];int ls[5000005],rs[5000005],sum[5000005];vector<int> id[N];inline int read(){    int a=0,f=1; char c=getchar();    while (c<'0'||c>'9') {if (c=='-') f=-1; c=getchar();}    while (c>='0'&&c<='9') {a=a*10+c-'0'; c=getchar();}    return a*f;}inline void insert(int x){    scanf("%s",s);    int c,now=1,len=strlen(s);    for (int i=len-1;~i;i--)    {        c=s[i]-'a';        if (a[now][c]) now=a[now][c];        else now=a[now][c]=++cnt;    }    id[now].push_back(x); p[x]=now; }void dfs(int x){    in[x]=++dfn; pos[dfn]=x;    for (int i=0;i<26;i++)        if (a[x][i]) dfs(a[x][i]);    out[x]=dfn;}void update(int l,int r,int x,int &y,int val){    y=++size;    ls[y]=ls[x]; rs[y]=rs[x]; sum[y]=sum[x]+1;    if (l==r) return;    int mid=l+r>>1;    if (val<=mid) update(l,mid,ls[x],ls[y],val);    else update(mid+1,r,rs[x],rs[y],val);}int query(int x,int y,int k){    if (k>sum[y]-sum[x]) return -1;    int l=1,r=n;    while (l!=r)    {        int mid=l+r>>1;        int tmp=sum[ls[y]]-sum[ls[x]];        if (tmp>=k) r=mid,x=ls[x],y=ls[y];        else l=mid+1,k-=tmp,x=rs[x],y=rs[y];    }    return l;}int main(){//  freopen("password.in","r",stdin);//  freopen("password.out","w",stdout);    n=read(); cnt=1;    for (int i=0;i<26;i++) a[0][i]=1;    for (int i=1;i<=n;i++) insert(i);    dfs(1);    for (int i=1;i<=cnt;i++)        if (id[pos[i]].size()!=0)         {            update(1,n,root[i-1],root[i],id[pos[i]][0]);            for (int j=1;j<id[pos[i]].size();j++)                 update(1,n,root[i],root[i],id[pos[i]][j]);        }        else root[i]=root[i-1];    for (int i=1;i<=n;i++)    {        int k=read();        printf("%d\n",query(root[in[p[i]]-1],root[out[p[i]]],k));    }    return 0;}