【HDOJ6223】Infinite Fraction Path（后缀数组，倍增）

题意：

给一个长度为n的字符串s[0..n-1]，但i的后继不再是i+1，而是(i*i+1)%n，求所有长度为n的“子串”中，字典序最大的是谁

n<=150000,s[i]=0..9

思路：后缀数组

因为前驱与后继的关系已经变化，就不能用下标直接加减

i的后继是唯一的，i的前驱却不一定

所以对于后继使用倍增，对于前驱每个位置暴力开队列存储，需要的时候再拿出来

在判断的地方稍作修改

 #include<cstdio>
 #include<cstring>
 #include<string>
 #include<cmath>
 #include<iostream>
 #include<algorithm>
 #include<map>
 #include<set>
 #include<queue>
 #include<vector>
 #include<bits/stdc++.h>
 using namespace std;
 typedef long long ll;
 typedef unsigned int uint;
 typedef unsigned long long ull;
 typedef pair<int,int> PII;
 typedef vector<int> VI;
 #define fi first
 #define se second
 #define MP make_pair
 #define N   310000
 #define MOD 1000000007
 #define eps 1e-8
 #define pi acos(-1)
 queue<int> q[N];

 char ch[N];
 int n,s[N],sa[N],wa[N],wb[N],wc[N],wd[N],height[N],Rank[N],
     f[N][];

 int read()
 {
    int v=,f=;
    char c=getchar();
    while(c<||<c) {if(c=='-') f=-; c=getchar();}
    while(<=c&&c<=) v=(v<<)+v+v+c-,c=getchar();
    return v*f;
 }

 bool cmp(int *r,int a,int b,int l)
 {
     return r[a]==r[b]&&r[f[a][l]]==r[f[b][l]];
 //    return r[a]==r[b]&&r[a+l]==r[b+l]; 原来的写法
 }

 void getsa(int *r,int *sa,int n,int m)
 {
 //    printf("\n");
     int *x=wa,*y=wb,p;
     int k=;
     for(int i=;i<n;i++) wc[x[i]=r[i]]++;
     for(int i=;i<m;i++) wc[i]+=wc[i-];
     for(int i=n-;i>=;i--) sa[--wc[x[i]]]=i;
     for(int j=,p=;p<n;j*=,m=p)
     {
         p=;
         /*for(i=n-j;i<n;i++) y[p++]=i;
         for(i=0;i<n;i++)
          if(sa[i]>=j) y[p++]=sa[i]-j;
         */ //默认后继为i+1应该是这个写法 

         for(int i=;i<n;i++) q[f[i][k]].push(i);
         for(int i=;i<n;i++)
          while(!q[sa[i]].empty())
          {
              y[p++]=q[sa[i]].front();
              q[sa[i]].pop();
          }
         for(int i=;i<n;i++) wd[i]=x[y[i]];
         for(int i=;i<m;i++) wc[i]=;
         for(int i=;i<n;i++) wc[wd[i]]++;
         for(int i=;i<m;i++) wc[i]+=wc[i-];
         for(int i=n-;i>=;i--) sa[--wc[wd[i]]]=y[i];
         swap(x,y);
         p=; x[sa[]]=;
         for(int i=;i<n;i++)
          x[sa[i]]=cmp(y,sa[i-],sa[i],k)?p-:p++;
         //printf("%d %d %d\n",k,j,p);
         k++;
         if(j>n) break; //加上这句就A了，应该都是N很大，D[i]相同的数据吧，名次的并列很多
     }
 }    

 int main()
 {
     freopen("hdoj6223.in","r",stdin);
     freopen("hdoj6223.out","w",stdout);
     int cas;
     scanf("%d",&cas);
     for(int v=;v<=cas;v++)
     {
         printf("Case #%d: ",v);
         scanf("%d",&n);
          scanf("%s",ch);
          for(int i=;i<n;i++) s[i]=ch[i]-''+;
          for(int i=;i<n;i++) f[i][]=(1LL*i*i+)%n;

         for(int j=;j<=;j++)
          for(int i=;i<n;i++) f[i][j]=f[f[i][j-]][j-]; 

          s[n]=;
          getsa(s,sa,n+,);
         int k=sa[n];
         for(int i=;i<=n;i++)
         {
             printf("%c",ch[k]);
             k=f[k][];
         }
         printf("\n");
         for(int i=;i<=max(,n);i++)
         {
                s[i]=sa[i]=wa[i]=wb[i]=wc[i]=wd[i]=
               Rank[i]=;
         }
     }
     return ;
 }