hdu4436-str2int（后缀数组 or 后缀自动机）
题意：给你一堆字符串，仅包含数字'0'到'9'。
例如 101 123
有一个字符串集合S包含输入的N个字符串，和他们的全部字串。
操作字符串很无聊，你决定把它们转化成数字。
你可以把一个字符串转换成一个十进制整数。
如果一个数字出现了多次，只留一个。
计算所有数字的和，模2012。
1<=N<=10000
题解：参考 http://blog.csdn.net/kqzxcmh/article/details/8122747
每个后缀的所有前缀就是所有子串。sa[i]和sa[i-1]有height[i]的长度是重复的，可以不考虑。
把字符串连起来，中间加分隔符，构成一个大的字符串，然后整体求后缀数组。
对于一个字符串123
可以v[0]=1 v[1]=12 v[3]=123 v[i]是0-i构成的数字
sum是前缀和 sum[0]=1 sum[1]=13 sum[2]=136
对于后缀23 sum[2]-sum[0] 得到 12+123的和
对于12需要减去10 对于123需要减去100
所以求l~r 首先需要sum[r]-sum[l-1], 然后设x=r-l
需要减去 for(int i=1;i<=x;++i) pow(10, i);
可以预处理出来。
太难辣！！像我这种辣鸡不适合这么难的题T^T
时限3s，187ms还是比较快的。
//后缀数组 #include <stdio.h> #include <cstring> #include <iostream> #include <algorithm> using namespace std; typedef long long ll; const int N = int(2e5)+10; const int M = 2012; int cmp(int *r,int a,int b,int l){ return (r[a]==r[b]) && (r[a+l]==r[b+l]); } // 用于比较第一关键字与第二关键字, // 比较特殊的地方是,预处理的时候,r[n]=0(小于前面出现过的字符) int wa[N],wb[N],wss[N],wv[N]; int sa[N]; // 排第i的是谁 i:1~n sa:0~n-1 int rk[N], // i排第几 i:0~n-1 rk:1~n height[N]; // 排名相邻的两个后缀的最长公共前缀长度:suffix(sa[i-1])和(sa[i]) 的最长公共前缀， char str[N]; void DA(char *r,int *sa,int n,int m){ // 此处N比输入的N要多1，为人工添加的一个字符，用于避免CMP时越界 m是不同的字符的个数 int i,j,p,*x=wa,*y=wb,*t; for(i=0;i<m;i++) wss[i]=0; for(i=0;i<n;i++) wss[x[i]=r[i]]++; for(i=1;i<m;i++) wss[i]+=wss[i-1]; for(i=n-1;i>=0;i--) sa[--wss[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]]; for(i=0;i<m;i++) wss[i]=0; for(i=0;i<n;i++) wss[wv[i]]++; for(i=1;i<m;i++) wss[i]+=wss[i-1]; for(i=n-1;i>=0;i--) sa[--wss[wv[i]]]=y[i]; 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++; } } void calheight(char *r,int *sa,int n){ // 此处N为实际长度 int i,j,k=0; for(i=1;i<=n;i++) rk[sa[i]]=i; for(i=0;i<n; height[rk[i++]] = k ) for(k?k--:0,j=sa[rk[i]-1]; r[i+k]==r[j+k]; k++); } char word[N]; int d[N]; // 该字符串的结束位置 int sum[N]; int v[N]; int Pow[N]; int cal(int l, int r) { if (l > r) return 0; //printf(">>%d %d %d %d %d\n", l, r, sum[r+1], sum[l], v[l]); //printf("%d\n", ((sum[r+1]-sum[l]) - v[l]*Pow[r-l+1]%M + M) % M); return ((sum[r+1]-sum[l]) - v[l]*Pow[r-l+1]%M + M) % M; } int main(int argc, char const *argv[]) { //freopen("in.txt", "r", stdin); Pow[0]=1;Pow[1]=10; for (int i=2;i<N;++i) Pow[i]=(Pow[i-1]+1)*10%M; int n; while (~scanf("%d", &n)) { int idx = 0; for (int i = 0; i < n; ++i) { scanf("%s", word); int len = strlen(word); int val = 0; for (int j = 0; j < len; ++j) { str[idx] = word[j] - '0' + 1; val = (val * 10 + str[idx] - 1) % M; v[idx+1] = val; //+1是因为会有-1 <0很麻烦= = sum[idx+1] = (sum[idx] + val) % M; d[idx] = len - j + idx; idx++; } d[idx] = idx; sum[idx+1] = v[idx+1] = 0; str[idx++] = 11; //1~11 } str[idx] = 0; DA(str, sa, idx+1, 15); calheight(str, sa, idx); int ans = 0; for (int i = 0; i < idx; ++i) { if (str[i] == 1 || str[i] == 11) continue; // 不要前导零 int l = i + height[rk[i]];//重复的部分 int r = d[i]-1; //printf("i=%d,l=%d, r=%d\n", i, l, r); if (l > r) continue; ans = (ans + cal(i, r) - cal(i, l-1)) % M; //ans = (ans + cal(l, r)) % M; 这个不对 因为所求字符串是从i开始的 ans = (ans + M) % M; } printf("%d\n", ans); } return 0; }
后缀自动机= = 先留坑吧orz。。。。
