[ccKILLKTH]Killjee and kth letter

建立后缀树(即反序插入字符串的parent树),然后可以发现按照dfs序排列满足其反串按字典序从小到大排列,那么就可以维护出每一刻子树的串长和,然后直接在dfs序上二分确定节点,再在节点内部乱搞即可求出答案。

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 #define N 200005
 4 #define ll long long
 5 int V,last,ans,a[N<<1],sz[N<<1],mi[N<<1],id[N<<1],len[N<<1],fa[N<<1],ch[N<<1][31];
 6 ll k,mod,l[N<<1],sum[N<<1];
 7 char s[N];
 8 void add(int c){
 9     int p=last,np=last=++V;
10     len[np]=len[p]+1;
11     sz[np]=1;
12     for(;(p)&&(!ch[p][c]);p=fa[p])ch[p][c]=V;
13     if (!p)fa[np]=1;
14     else{
15         int q=ch[p][c];
16         if (len[q]==len[p]+1)fa[np]=q;
17         else{
18             int nq=++V;
19             len[nq]=len[p]+1;
20             memcpy(ch[nq],ch[q],sizeof(ch[q]));
21             fa[nq]=fa[q];
22             fa[np]=fa[q]=nq;
23             for(;(p)&&(ch[p][c]==q);p=fa[p])ch[p][c]=nq;
24         }
25     }
26 }
27 void dfs(int k){
28     a[++a[0]]=k;
29     for(int i=0;i<26;i++)
30         if (ch[k][i])dfs(ch[k][i]);
31 }
32 int main(){
33     int t;
34     scanf("%s%d",s,&t);
35     V=last=1;
36     int n=strlen(s);
37     for(int i=n-1;i>=0;i--)add(s[i]-'a');
38     for(int i=1;i<=V;i++)a[len[i]]++;
39     for(int i=1;i<=n;i++)a[i]+=a[i-1];
40     for(int i=1;i<=V;i++)id[a[len[i]]--]=i;
41     for(int i=1,j=n;i<=V;i++)
42         if (sz[i])mi[i]=--j;
43     memset(a,0,sizeof(a));
44     for(int i=V;i;i--){
45         sz[fa[id[i]]]+=sz[id[i]];
46         mi[fa[id[i]]]=mi[id[i]];
47     }
48     for(int i=V;i;i--)l[i]=(len[fa[i]]+len[i]+1LL)*(len[i]-len[fa[i]])/2*sz[i];
49     memset(ch,0,sizeof(ch));
50     for(int i=1;i<=V;i++)ch[fa[i]][s[mi[i]+len[fa[i]]]-'a']=i;
51     dfs(1);
52     for(int i=1;i<=V;i++)sum[i]=sum[i-1]+l[a[i]];
53     while (t--){
54         scanf("%lld%lld",&k,&mod);
55         k=k*ans%mod+1;
56         int p=lower_bound(sum+1,sum+V+1,k)-sum;
57         k-=sum[p-1];
58         p=a[p];
59         int x=len[fa[p]]+1,y=len[p];
60         while (x<y){
61             int mid=(x+y>>1);
62             if ((mid+len[fa[p]]+1LL)*(mid-len[fa[p]])/2*sz[p]>=k)y=mid;
63             else x=mid+1;
64         }
65         k-=(x+len[fa[p]])*(x-1LL-len[fa[p]])/2*sz[p];
66         printf("%c\n",s[(k-1)%x+mi[p]]);
67         ans+=s[(k-1)%x+mi[p]];
68     }
69 }
View Code
原文地址:https://www.cnblogs.com/PYWBKTDA/p/11254575.html