【模板】后缀数组

虽然基本是照着打的。。。感觉还是很好。。。

#include<iostream>
#include<cstdio>
#include<cstring>
#define MAXN 300000 
using namespace std;

int n,m,sa[MAXN],rank[MAXN],h[MAXN];
int _x[MAXN],_y[MAXN],tx[MAXN],bit[MAXN];

long long ans=0;
///*
int s[MAXN];
/*
void into()
{
  scanf("%d",&n);
  getchar();
  for(int i=0;i<=n;i++)
    scanf("%c",&s[i]); 
  s[n++]='a'-1;
  for(int i=0;i<n;i++)
    _x[i]=s[i]-'a'+1;    
}
//*/
void into() 
{
  scanf("%d",&n);
  char t;int i;
  for (i=0;i<n;i++) 
  {
    for (t=getchar();!(t>='a'&&t<='z');t=getchar()) ;
      s[i]=t;
  }
  s[n++]='a'-1;
  for(i=0;i<n;i++)
    _x[i]=s[i]-'a'+1; 
}

void print()
{
  printf("\ns:");
  for(int i=0;i<n;i++)
    printf("    %c",s[i]);
  printf("\nsa:");
  for(int i=0;i<n;i++)
    printf("    %d",sa[i]);
  printf("\nrank:");
  for(int i=0;i<n;i++)
    printf("    %d",_x[i]);
  printf("\nh:");
  for(int i=0;i<n;i++)
    printf("    %d",h[i]);
}

inline int cmp(int *r,int op,int ed,int l)
{return (r[op]==r[ed])&&(r[op+l]==r[ed+l]);}

void da()
{
  m=128;
  int *x=_x,*y=_y,*t;
  int i,j,p;
  for(i=0;i<m;i++)bit[i]=0;
  for(i=0;i<n;i++)bit[x[i]]++;
  for(i=1;i<m;i++)bit[i]+=bit[i-1];
  for(i=n-1;i>=0;i--)sa[--bit[x[i]]]=i;
  
  for(p=1,j=1;p<n;j<<=1,m=p)
  {
    for(p=0,i=n-j;i<n;i++)y[p++]=i;
    for(i=0;i<n;i++)if(sa[i]>=j)y[p++]=sa[i]-j;
    
    for(i=0;i<n;i++)tx[i]=x[y[i]];
    
    for(i=0;i<m;i++)bit[i]=0;
    for(i=0;i<n;i++)bit[tx[i]]++;
    for(i=1;i<m;i++)bit[i]+=bit[i-1];
    for(i=n-1;i>=0;i--)sa[--bit[tx[i]]]=y[i];
    
    t=x,x=y,y=t;
    p=1,x[sa[0]]=0;
    for(i=1;i<n;i++)
      x[sa[i]]=cmp(y,sa[i-1],sa[i],j) ? p-1:p++;       
  }
  for(i=1;i<n;i++)
    rank[sa[i]]=i;
}

void cal()
{
  int k=0,i,j;
  for(i=0;i<n-1;h[rank[i++]]=k)
    for(k?k--:0,j=sa[rank[i]-1];s[i+k]==s[j+k];k++);     
}

void work()
{
  for(int i=0;i<n;i++)
    ans+=(i-h[i]);
  printf("%lld",ans);
}

int main()
{
  into(); 
  //print(); 
  da();
  cal();
  //print();  
  work();  
  
    
    
    
  //while(1);
  return 0;    
}