hdu 4436 后缀数组

明显的与后缀数组神马的有关，添加不同的分隔符连起来，用后缀数组模板搞一下。

然后就是计数的问题了。

对于每一个后缀，要看它所能贡献的前缀。比如1234￥...... 就可以贡献1 12 123 1234

如果说对于第i个后缀，height【ranks【i】】>0，说明有些已经重复，要删去重复的。

然后暴力加和。。果然T了。

根据加和的规律性可以进行优化：

例如:

 t[1]=1   t[2]=1+12   t[3]=1+12+123   t[4]=1+12+123+1234

那么2+23+234=t[4]-t[1]-1*1110

同理 3+34=t[4]-t[2]-12*110

另外今天才知道c语言中负数取余等于绝对值取余再加负号TAT

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

const int N = 120000;
const int mod = 2012;

int n;

int r[N], tx[N], ty[N], rs[N], ranks[N], sa[N], height[N], rmq[N][20]; //rs基数排序
int T[N],V[N],POW[N];
char ts[N];

bool cmp(int *r, int a, int b, int len)
{
	return (r[a] == r[b]) && (r[a + len] == r[b + len]);
}

void suffix(int n, int m) //n为长度，最大值小于m
{
	int i, j, p, *x = tx, *y = ty, *t;
	for(i = 0; i < m; ++i)
		rs[i] = 0;
	for(i = 0; i < n; ++i) {
		x[i] = r[i];
		++rs[x[i]];
	}
	for(i = 1; i < m; ++i)
		rs[i] += rs[i - 1];
	for(i = n - 1; i >= 0; --i)
		sa[--rs[x[i]]] = i;
	for(j = p = 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 < m; ++i)
			rs[i] = 0;
		for(i = 0; i < n; ++i)
			++rs[x[y[i]]];
		for(i = 1; i < m; ++i)
			rs[i] += rs[i - 1];
		for(i = n - 1; i >= 0; --i)
			sa[--rs[x[y[i]]]] = y[i];
		t = x, x = y, y = t;
		for(i = 1, p = 1, x[sa[0]] = 0; i < n; ++i) {
			if(cmp(y, sa[i - 1], sa[i], j))
				x[sa[i]] = p - 1;
			else
				x[sa[i]] = p++;
		}
	}
	/**for(i = 0; i < n; ++i)
	printf("%s\n", s + sa[i]);*/
}

void calheight(int n)
{
	int i, j, k = 0;
	for(i = 1; i <= n; ++i)
		ranks[sa[i]] = i;
	for(i = 0; i < n; ++i) {
		if(k)
			--k;
		j = sa[ranks[i] - 1];
		while(r[i + k] == r[j + k])
			++k;
		height[ranks[i]] = k;
	}
}


int main()
{
	POW[1]=10;
	for(int i=2;i<110000;i++)  
        POW[i]=(POW[i-1]+1)*10%mod; 
	while( scanf("%d", &n)!=EOF)
	{
		int i,j,p,q,w,k=0,l1,ans=0,temp,ta,tb,tc;
		for(i=1;i<=n;i++)
		{
			scanf("%s",ts);
			l1=strlen(ts);
			for(q=0,p=k;q<l1;q++,p++)
				r[p]=ts[q];
			k+=l1;
			r[k++]=i+200;
		}
		r[k]=0;
		suffix(k + 1,15000);
		calheight(k);
		ans = 0;

		T[0]=0;
		V[0]=0;
		p=1;
		q=0;
		temp=0;
		for(i=q;i<k;i++)
		{
			if(r[i]>=200)
			{
				for(j=q,w=1;j<i;j++,w++)
				{
					if(r[j]==48)
						continue;
					ta=(T[p-1]-T[w-1]-V[w-1]*POW[p-w]%mod)%mod;
					//while(ta<0)
					//	ta+=mod;
					tb=(T[w-1+height[ranks[j]]]-T[w-1]-V[w-1]*POW[height[ranks[j]]])%mod;
					//while(tb<0)
					//	tb+=mod;
					tc=ta-tb;
					//while(tc<0)
					//	tc+=mod;
					ans+=tc;
					ans%=mod;
				}
				T[0]=0;
				V[0]=0;
				p=1;
				temp=0;
				q=i+1;
				i=q;
			}
			temp=(temp*10+r[i]-48)%mod;
			T[p]=(T[p-1]+temp)%mod;
			V[p]=temp;
			p++;	
		}
		while(ans<0)
			ans+=mod;
		printf("%d\n", ans%mod);
	}
	return 0;
}