HDU5769-Substring-多校#4-1006-后缀数组

给定一个字符x和一个字符串。要求输出包含此字符的所有不同字串。

后缀数组可以计算一个字符串的所有不同字串，理解了原理就能做这题了。

对于每一个后缀i，将产生len-sa[i]-hight[i]的前缀，累加起来就是所有不同字串。这里要求字串必须包含x

也就是对于每一个后缀i，要减去不含x的前缀。

保存每一个x的位置，用lower_bound查询在此后缀中的第一个x，然后之前的都不计算到贡献当中。

dc3敲起来太慢了

 #include <cstdio>
 #include <algorithm>
 #include <cstring>
 #define LL long long
 //using namespace std;

 const int maxn = 1e5+;
 #define F(x) ((x)/3+((x)%3 == 1?0:tb))
 #define G(x) ((x)<tb?(x)*3+1:((x)-tb)*3+2)
 int wa[maxn*],wb[maxn*],wv[maxn*],wss[maxn*];

 int c0(int *r,int a,int b)
 {
     return r[a] == r[b] && r[a+] == r[b+] && r[a+] == r[b+];
 }
 int c12(int k,int *r,int a,int b)
 {
     if(k == )
         return r[a] < r[b] ||(r[a] == r[b] && c12(,r,a+,b+));
     else
         return r[a] < r[b] ||(r[a] == r[b] && wv[a+] < wv[b+]);
 }
 void sort(int *r,int *a,int *b,int n,int m)
 {
     int i;
     for(i=;i<n;i++) wv[i] = r[a[i]];
     for(i=;i<m;i++) wss[i] = ;
     for(i=;i<n;i++) wss[wv[i]]++;
     for(i=;i<m;i++) wss[i] += wss[i-];
     for(i=n-;i>=;i--)b[--wss[wv[i]] ] = a[i];
 }
 void dc3(int *r,int *_sa,int n,int m)
 {
     int i,j,*rn = r+n;
     int *san = _sa+n,ta = ,tb=(n+)/,tbc=,p;
     r[n] = r[n+] = ;
     for(i=;i<n;i++) if(i% != ) wa[tbc++] = i;
     sort(r+,wa,wb,tbc,m);
     sort(r+,wb,wa,tbc,m);
     sort(r,wa,wb,tbc,m);
     for(p=,rn[F(wb[])] = ,i=;i<tbc;i++)
         rn[F(wb[i])] = c0(r,wb[i-],wb[i]) ? p-:p++;
     if(p < tbc) dc3(rn,san,tbc,p);
     else for(i=;i<tbc;i++) san[rn[i]] = i;
     for(i=;i<tbc;i++) if(san[i]<tb) wb[ta++] = san[i]*;
     if(n%==) wb[ta++] = n-;
     sort(r,wb,wa,ta,m);
     for(i=;i<tbc;i++) wv[wb[i]=G(san[i]) ] = i;
     for(i=,j=,p=;i<ta&&j<tbc;p++)
         _sa[p] = c12(wb[j]%,r,wa[i],wb[j]) ? wa[i++]:wb[j++];
     for(;i<ta;p++) _sa[p] = wa[i++];
     for(;j<tbc;p++) _sa[p] = wb[j++];
 }
 void da(int str[],int _sa[],int _rank[],int _height[],int n,int m)
 {
     for(int i=n;i<n*;i++) str[i] = ;
     dc3(str,_sa,n+,m);
     int i,j,k = ;
     for(i=;i<=n;i++) _rank[_sa[i]] = i;
     for(i=;i<n;i++)
     {
         if(k) k--;
         j = _sa[_rank[i]- ];
         while(str[i+k] == str[j+k]) k++;
         _height[_rank[i]] = k;
     }
 }

 int T;
 int sa[*maxn],hehe[*maxn],rank[maxn],height[maxn];
 int x;
 char line[*maxn];

 void debug(int n)
 {
     for(int i=;i<=n;i++)
     {
         printf("sa:%d rank:%d height:%d\n",sa[i],rank[i],height[i]);
     }
 }

 int main()
 {
     //freopen("input.txt","r",stdin);
     scanf("%d ",&T);
     int cas = ;
     while(T--)
     {
         scanf(" %c %s",&x,line);
         int len = strlen(line);
         for(int i=;i<len;i++) hehe[i] = line[i];
         int ch[maxn],cnt = ;
         for(int i=;i<len;i++)
         {
             if(line[i] == x) ch[cnt++] = i;
         }
         ch[cnt++] = len;
         da(hehe,sa,rank,height,len,);
         //debug(len);
         LL ans = ;
         for(int i=;i<=len;i++)
         {
             int pos = ch[std::lower_bound(ch,ch+cnt,sa[i]) - ch] - sa[i];
             ans += len -sa[i]  ;
             ans -= std::max(pos,height[i]);
         }
         printf("Case #%d: %I64d\n",++cas,ans);
     }
 }