HDU 4436 str2int 后缀数组 字符串哈希 前缀和

题意：给出N个数字序列，每个序列的子串都能组成数字。求出这些序列所有子串对应的数字的和。对于相同的数字，我们只考虑一次。忽略前导零。

思路：（以前都是做单个字符串的题，第一次做多个字符串的题，跪了。。。只知道思想，不写出代码还是不会呀。下面是看网上题解才明白的。）

           我们可以用后缀数组求出所有的不相同的子串。但是我们怎么求出对应字符串的和呢？

           这里就用到了字符串哈希的思想。我们可以求出到第i位，1-i位数字的值（即前缀和），然后我们再求出前缀和的前缀和。同时，我们记录进制数的前缀和。这样，我们就O（1）求出任意[l,r]内对应子串能产生的所有数字的和。

代码来自http://blog.csdn.net/kqzxcmh/article/details/8122747

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <vector>
#include <map>
#include <queue>
#define maxn 300000
#define MOD 2012
using namespace std;

int wa[maxn],wb[maxn],wv[maxn];
int rank[maxn],height[maxn],wS[maxn],sa[maxn],P[maxn],Pow[maxn],V[maxn],
T[maxn],Len[maxn],S[maxn],arrive[maxn],n,len,value;
char s[maxn];
int cmp(int *r,int a,int b,int l) {return r[a]==r[b]&&r[a+l]==r[b+l];}
void da(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]];
         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];
         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;
}


void calheight(int *r,int *sa,int n)
{
     int i,j,k=0;
     for(i=0;i<n;i++) rank[sa[i]]=i;
     for(i=0;i<n;height[rank[i++]]=k)
     if(rank[i])for(k?k--:0,j=sa[rank[i]-1];r[i+k]==r[j+k];k++);
     return;
}
int  get(int l,int r)
{
     if(l>r) return 0;
     int ans=(T[r+1]-T[l])%MOD;
     //ans-=V[l]*(Pow[r]-(l?Pow[l-1]:0));
     ans-=V[l]*(Pow[r-l+1]);
     ans%=MOD;
     if (ans<0) ans+=MOD;
     return ans;
}
int main()
{
    Pow[0]=1;Pow[1]=10;
    for (int i=2;i<200000;++i) 
        Pow[i]=(Pow[i-1]+1)*10%MOD;
    while (scanf("%d",&n)!=EOF)
    {     
          len = 0; value = 0;
          for (int i=0;i < n ;++i){
              scanf("%s",s);
              Len[i]= strlen(s);
              for (int j = 0 ; j < Len[i]; ++j){
                  S[len]=s[j]-'0'+1;
                  value = (value*10+S[len]-1)%MOD;
                  V[len+1]=value;
                  T[len+1]=(T[len]+value)%MOD;
                  P[len]=i;
                  len++;
              }
                  S[len]=11;
                  value = (value*10+10)%MOD;
                  V[len+1]=value;
                  T[len+1]=(T[len]+value)%MOD;
                  P[len]=i;
                  len++;
                  arrive[i]=len-2;
          }
          S[len-1]=0;
          da(S,sa,len,12);
          //for (int i=0;i<len;++i) cout << sa[i]<<" ";cout<<endl;
          calheight(S,sa,len);
         //for (int i=0;i<len;++i) cout << rank[i]<<" ";cout<<endl;
          //cout<<len<<endl;
          int ans = 0;
          for (int  i = 0;i < len;++i)
          if (S[i]%10 !=1){
              if (i+height[rank[i]]<= arrive[P[i]]){
               ans+=get(i,arrive[P[i]])-get(i,i+height[rank[i]]-1); 
               ans%=MOD;if (ans<0) ans+=MOD;           
               }
          }
          cout << ans << endl;
    }
    return 0;
}

学姿势：

1.在字符串中求值，注意哈希的使用。

2.多个字符串拼成一个，记录位置。

3.前缀和