HDU str2int 前缀和的力量

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;

const int maxn=200050;
const int mod=2012;

//以下为倍增算法求后缀数组
int wa[maxn],wb[maxn],wv[maxn],Ws[maxn];
int cmp(int *r,int a,int b,int l)
{return r[a]==r[b]&&r[a+l]==r[b+l];}

void da(const int *r,int *sa,int n,int m){
    int i,j,p,*x=wa,*y=wb,*t; 
    for(i=0;i<m;i++) Ws[i]=0; 
    for(i=0;i<n;i++) Ws[x[i]=r[i]]++; 
    for(i=1;i<m;i++) Ws[i]+=Ws[i-1]; 
    for(i=n-1;i>=0;i--) sa[--Ws[x[i]]]=i; 
    for(j=1,p=1;p<n;j*=2,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++) wv[i]=x[y[i]]; //x[]->上次排序的排名
        for(i=0;i<m;i++) Ws[i]=0; 
        for(i=0;i<n;i++) Ws[wv[i]]++; 
        for(i=1;i<m;i++) Ws[i]+=Ws[i-1]; 
        for(i=n-1;i>=0;i--) sa[--Ws[wv[i]]]=y[i]; 
        //y[i]->二级排序后第i大的下标，循环从n-1到0由于一级排序相同要看二级排序
        for(t=x,x=y,y=t,p=1,x[sa[0]]=0,i=1;i<n;i++) 
            x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++; 
    } 
    return; 
}
int sa[maxn],Rank[maxn],height[maxn];
//求height数组
//sa[1,n]取值范围[0,n-1],sa[0]为特殊字符
//Rank[0,n-1]取值范围[1,n]
//height[1,n]表示排名相邻的两个后缀的最长公共前缀
//height[i]表示sa[i]和sa[i-1]的最长前缀 排名为i与i-1的最长公共前缀
void calheight(const int *r,int *sa,int n){
    int i,j,k=0;
    for(i=1;i<=n;i++) Rank[sa[i]]=i;
    for(i=0;i<n;height[Rank[i++]]=k)
        for(k?k--:0,j=sa[Rank[i]-1];r[i+k]==r[j+k];k++);
    return;
}
/*
da(r,sa,n+1,128);
n+1表示加入特殊字符后的长度[0,n];
calheight(r,sa,n);
去掉特殊字符[1,n]长度，就是输入字符串长度n
*/

int pos[maxn];
char ch[maxn];
int a[maxn];
int dig[maxn];
int Pow[maxn];
int lsum[maxn];
int sum[maxn];
int ID[maxn];
int p[maxn];

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

int Min(int a,int b){
	return a<b?a:b;
}
int Gao(int n){
    int i,j,k,l,r,mid;
    int id=0,ret=0;
    memset(lsum,0,sizeof(lsum));
    memset(sum,0,sizeof(sum));
    while(id<n){
        l=id;r=ID[id]-1;
        lsum[l]=dig[l];
        for(i=id+1;i<=r;i++){
            lsum[i]=(lsum[i-1]*10+dig[i])%mod;//求前缀数
        }
        sum[r]=lsum[r];
        for(i=r-1;i>=l;i--){
            sum[i]=(sum[i+1]+lsum[i])%mod;//求前缀和
        }
        id=ID[id];
    }
    for(i=1;i<=n;i++){
        l=sa[i];r=ID[sa[i]]-1;
        if(!dig[sa[i]])continue;
        mid=height[i];
        if(mid+l-1>r)mid=r-l+1;
        int x=l+mid;
		if(x<=r){
			ret=(ret+sum[x])%mod;
		}
		if(l-1>=pos[l] && x<=r){
			/*
				后缀的前缀就是子串
				st 表示原串开始位置 ed原串终点位置
				st l  x  ed 
				我们要求[l,ed]去掉重复的，即前面求过的[l,x]即ret+sum[x],
				在减去多算的[st,l-1],值t=lsum[l-1]*p[x-1],Pow[r-x+1]表示t的10^k的多算个数
			*/
			ret=(ret-(lsum[l-1]*p[x-l]%mod)*Pow[r-x+1])%mod;//
		}
    }
    return (ret+mod)%mod;
}

int main(){
	int dif;
    int n,m;
    int i,j,k;
    init();
    while(scanf("%d",&m)!=EOF && m){
        n=0;
		dif=12;
/*
不加dif，是字符串隔开，无法按字典排序去重如
a ->111
b ->11
组成一个111 11字符串
a 1
a 11
b 1 a 11
b 11 a 11
b 111 a 11
  */
		memset(dig,0,sizeof(dig));
        for(i=0;i<m;i++){
            scanf("%s",ch);
            int len=strlen(ch);
            int cur=n;
            for(j=0;j<len;j++){
                a[n]=ch[j]-'0'+1;
                ID[n]=len+cur;
				pos[n]=cur;
                dig[n++]=ch[j]-'0';
            }
			pos[n]=n;
			ID[n]=n+1;
			a[n++]=dif++;
        }
		ID[n]=n+1;pos[n]=n;
        a[n]=0;
        da(a,sa,n+1,dif);
        calheight(a,sa,n);
        printf("%d\n",Gao(n));
    }
    return 0;
}


/*
2
111
11


6
000000000001122
25454512549549483
12165415151
545674874231
242142
245454456485745331313343
  */