hdu 4436 str2int 后缀数组

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <set>
using namespace std;
const int maxn=1e5+1e5+9,mod=2012;
char a[maxn];
int r[maxn];
int next[maxn];
long long dp[maxn];
int prel[maxn],sum[maxn],pp[maxn],ff[maxn];
int minrmq[maxn][20];
int *rank,height[maxn],sa[maxn];
int wx[maxn],wy[maxn],cnt[maxn];//后缀数组使用的过程数组
//height排名为i和i-1的后缀的最长前缀，rank后缀的排名，sa排名为i的后缀的起始位置。
inline bool cmp(int *r,int a,int b,int l)
{
return r[a]==r[b]&&r[a+l]==r[b+l];
}
//n为字符串长度，m为最大的字符的值。
void da(int *r,int n,int m)//r数组不能出现0
{
r[n+1]=0;
int i,l,p,*x=wx,*y=wy,*t;
memset(cnt,0,sizeof(int)*(m+1));
for(int i=1;i<=n;i++) cnt[x[i]=r[i]]++;
for(int i=1;i<=m;i++) cnt[i]+=cnt[i-1];
for(int i=n;i>=1;i--) sa[cnt[x[i]]--]=i;
for(l=1,p=1;p<n;l<<=1,m=p)
{
for(p=1,i=n-l+1;i<=n;i++) y[p++]=i;
for(i=1;i<=n;i++)
if(sa[i]>l)
y[p++]=sa[i]-l;
memset(cnt,0,sizeof(int)*(m+1));
for(i=1;i<=n;i++) cnt[x[i]]++;
for(i=1;i<=m;i++) cnt[i]+=cnt[i-1];
for(i=n;i>=1;i--) sa[cnt[x[y[i]]]--]=y[i];
for(t=x,x=y,y=t,p=1,x[sa[1]]=1,i=2;i<=n;i++)
x[sa[i]]=cmp(y,sa[i-1],sa[i],l)?p:++p;
}
rank=x;
int j,k=0;
for(i=1;i<=n;height[rank[i++]]=k)
for(k?k--:0,j=sa[rank[i]-1];r[i+k]==r[j+k];k++);
return;
}

void getrmq(int n)//n为数据大小
{
for(int i=1;i<=n;i++)
minrmq[i][0]=height[i];
for(int i=1;(1<<i)<=n;i++)
for(int j=1;j+(1<<i)-1<=n;j++)
minrmq[j][i]=min(minrmq[j][i-1],minrmq[j+(1<<i-1)][i-1]);
}

int askrmq(int t,int s)
{
int k=log((double)s-t+1)/log((double)2);
int mmin=min(minrmq[t][k],minrmq[s-(1<<k)+1][k]);
return mmin;
}

void getpre(int n)
{
set <int> s;
s.clear();
for(int i=1;i<=n;i++)
{
set <int> :: iterator p=s.lower_bound(rank[n-i+1]);

if(p!=s.end())
{
prel[i]=askrmq(rank[n-i+1]+1,*p);
}

if(p!=s.begin())
{
p--;
prel[i]=max(prel[i],askrmq((*p)+1,rank[n-i+1]));
}
s.insert(rank[n-i+1]);
}

next[1]=1;
for(int i=2;i<=n;i++)
{
if(a[i-1]=='0')
{
next[i]=next[i-1];
}
else
next[i]=i;
}

//    for(int i=1;i<=n;i++)
//    {
//        if(prel[i]!=0)
//        prel[i]+=i-prel[i]+1-next[i-prel[i]+1];
////        else
////        prel[i]+=i-next[i]+1;
//    }

}

int zero[maxn];
void solve(int n)
{
dp[0]=0;
sum[0]=0;
ff[0]=0;

bool flag=false;
a[0]='#';
for(int i=0,k=1,m=(a[i]!=0);i<=n;i++)
{
if(a[i]=='#')
{
zero[i]=0;
sum[i]=0;
dp[i]=dp[i-1];
k=0;
m=0;
ff[i]=0;
flag=false;
while(a[i+1]=='0')
{
ff[i]=0;
sum[i]=0;
i++;
dp[i]=dp[i-1];
zero[i]=0;
}
}
else
{
k++;
sum[i]=(sum[i-1]*10+(a[i]-'0')*(k-m))%mod;
ff[i]=(ff[i-1]*10+a[i]-'0')%mod;
dp[i]=(dp[i-1]+sum[i])%mod;

int tmp=(ff[i]-ff[i-prel[i]]*pp[prel[i]]+mod*mod)%mod;

dp[i]-=sum[i]-(sum[i-prel[i]]*pp[prel[i]]+tmp*(k-prel[i]-zero[i-prel[i]]));
dp[i]=(dp[i]+(long long)2012000000*10)%mod;

zero[i]=zero[i-1];
if(a[i]=='0')
{
m++;
zero[i]++;
}
//            else m=0;

}
}
cout<<dp[n-1]<<endl;
}

void getpp()
{
pp[0]=1;
for(int i=1;i<maxn;i++)
pp[i]=(pp[i-1]*10)%mod;
}

int main()
{
freopen("in.txt","r",stdin);
int T;
while(scanf("%d",&T)!=EOF)
{
memset(prel,0,sizeof(prel));

int n=0;
char txt;
getchar();
for(int i=1;i<=T;i++)
{
while(1)
{
txt=getchar();
if(txt==' ') continue;
if(txt=='0') continue;
break;
}
while(1)
{
if(txt=='\n') break;
if(txt==' ')
{
txt=getchar();
continue;
}
a[++n]=txt;
txt=getchar();
}
a[++n]='#';
}
for(int i=1,t=0;i<=n;i++)
{
if(a[n-i+1]!='#')
r[i]=a[n-i+1];
else
{
t++;
r[i]='9'+t;
}
}
da(r,n,'9'+T);
getrmq(n);
getpre(n);
getpp();
solve(n);

//        for(int i=1;i<=n;i++)
//        cout<<prel[i]<<endl;

}
return 0;
}