【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][19];
    

int read()
{ 
   int v=0,f=1;
   char c=getchar();
   while(c<48||57<c) {if(c=='-') f=-1; c=getchar();}
   while(48<=c&&c<=57) v=(v<<3)+v+v+c-48,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("
");
    int *x=wa,*y=wb,p;
    int k=0; 
    for(int i=0;i<n;i++) wc[x[i]=r[i]]++;
    for(int i=1;i<m;i++) wc[i]+=wc[i-1];
    for(int i=n-1;i>=0;i--) sa[--wc[x[i]]]=i;
    for(int j=1,p=1;p<n;j*=2,m=p)
    {
        p=0;
        /*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=0;i<n;i++) q[f[i][k]].push(i);
        for(int i=0;i<n;i++)
         while(!q[sa[i]].empty())
         {
             y[p++]=q[sa[i]].front();
             q[sa[i]].pop();
         }
        for(int i=0;i<n;i++) wd[i]=x[y[i]];
        for(int i=0;i<m;i++) wc[i]=0;
        for(int i=0;i<n;i++) wc[wd[i]]++; 
        for(int i=1;i<m;i++) wc[i]+=wc[i-1];
        for(int i=n-1;i>=0;i--) sa[--wc[wd[i]]]=y[i];
        swap(x,y);
        p=1; x[sa[0]]=0;
        for(int i=1;i<n;i++) 
         x[sa[i]]=cmp(y,sa[i-1],sa[i],k)?p-1:p++;
        //printf("%d %d %d
",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=1;v<=cas;v++)
    {
        printf("Case #%d: ",v);
        scanf("%d",&n);
         scanf("%s",ch);
         for(int i=0;i<n;i++) s[i]=ch[i]-'0'+1;
         for(int i=0;i<n;i++) f[i][0]=(1LL*i*i+1)%n;
     
        for(int j=1;j<=18;j++)
         for(int i=0;i<n;i++) f[i][j]=f[f[i][j-1]][j-1]; 
    
         
         
         s[n]=0;
         getsa(s,sa,n+1,20);
        int k=sa[n];
        for(int i=1;i<=n;i++)
        {
            printf("%c",ch[k]);
            k=f[k][0];
        }
        printf("
");
        for(int i=0;i<=max(20,n);i++)
        {
               s[i]=sa[i]=wa[i]=wb[i]=wc[i]=wd[i]=
              Rank[i]=0; 
        } 
    }
    return 0;
}