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

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

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

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

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

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

dc3敲起来太慢了

  1 #include <cstdio>
  2 #include <algorithm>
  3 #include <cstring>
  4 #define LL long long
  5 //using namespace std;
  6
  7 const int maxn = 1e5+100;
  8 #define F(x) ((x)/3+((x)%3 == 1?0:tb))
  9 #define G(x) ((x)<tb?(x)*3+1:((x)-tb)*3+2)
 10 int wa[maxn*3],wb[maxn*3],wv[maxn*3],wss[maxn*3];
 11
 12 int c0(int *r,int a,int b)
 13 {
 14     return r[a] == r[b] && r[a+1] == r[b+1] && r[a+2] == r[b+2];
 15 }
 16 int c12(int k,int *r,int a,int b)
 17 {
 18     if(k == 2)
 19         return r[a] < r[b] ||(r[a] == r[b] && c12(1,r,a+1,b+1));
 20     else
 21         return r[a] < r[b] ||(r[a] == r[b] && wv[a+1] < wv[b+1]);
 22 }
 23 void sort(int *r,int *a,int *b,int n,int m)
 24 {
 25     int i;
 26     for(i=0;i<n;i++) wv[i] = r[a[i]];
 27     for(i=0;i<m;i++) wss[i] = 0;
 28     for(i=0;i<n;i++) wss[wv[i]]++;
 29     for(i=1;i<m;i++) wss[i] += wss[i-1];
 30     for(i=n-1;i>=0;i--)b[--wss[wv[i]] ] = a[i];
 31 }
 32 void dc3(int *r,int *_sa,int n,int m)
 33 {
 34     int i,j,*rn = r+n;
 35     int *san = _sa+n,ta = 0,tb=(n+1)/3,tbc=0,p;
 36     r[n] = r[n+1] = 0;
 37     for(i=0;i<n;i++) if(i%3 != 0) wa[tbc++] = i;
 38     sort(r+2,wa,wb,tbc,m);
 39     sort(r+1,wb,wa,tbc,m);
 40     sort(r,wa,wb,tbc,m);
 41     for(p=1,rn[F(wb[0])] = 0,i=1;i<tbc;i++)
 42         rn[F(wb[i])] = c0(r,wb[i-1],wb[i]) ? p-1:p++;
 43     if(p < tbc) dc3(rn,san,tbc,p);
 44     else for(i=0;i<tbc;i++) san[rn[i]] = i;
 45     for(i=0;i<tbc;i++) if(san[i]<tb) wb[ta++] = san[i]*3;
 46     if(n%3==1) wb[ta++] = n-1;
 47     sort(r,wb,wa,ta,m);
 48     for(i=0;i<tbc;i++) wv[wb[i]=G(san[i]) ] = i;
 49     for(i=0,j=0,p=0;i<ta&&j<tbc;p++)
 50         _sa[p] = c12(wb[j]%3,r,wa[i],wb[j]) ? wa[i++]:wb[j++];
 51     for(;i<ta;p++) _sa[p] = wa[i++];
 52     for(;j<tbc;p++) _sa[p] = wb[j++];
 53 }
 54 void da(int str[],int _sa[],int _rank[],int _height[],int n,int m)
 55 {
 56     for(int i=n;i<n*3;i++) str[i] = 0;
 57     dc3(str,_sa,n+1,m);
 58     int i,j,k = 0;
 59     for(i=1;i<=n;i++) _rank[_sa[i]] = i;
 60     for(i=0;i<n;i++)
 61     {
 62         if(k) k--;
 63         j = _sa[_rank[i]-1 ];
 64         while(str[i+k] == str[j+k]) k++;
 65         _height[_rank[i]] = k;
 66     }
 67 }
 68
 69 int T;
 70 int sa[3*maxn],hehe[3*maxn],rank[maxn],height[maxn];
 71 int x;
 72 char line[3*maxn];
 73
 74 void debug(int n)
 75 {
 76     for(int i=0;i<=n;i++)
 77     {
 78         printf("sa:%d rank:%d height:%d\n",sa[i],rank[i],height[i]);
 79     }
 80 }
 81
 82 int main()
 83 {
 84     //freopen("input.txt","r",stdin);
 85     scanf("%d ",&T);
 86     int cas = 0;
 87     while(T--)
 88     {
 89         scanf(" %c %s",&x,line);
 90         int len = strlen(line);
 91         for(int i=0;i<len;i++) hehe[i] = line[i];
 92         int ch[maxn],cnt = 0;
 93         for(int i=0;i<len;i++)
 94         {
 95             if(line[i] == x) ch[cnt++] = i;
 96         }
 97         ch[cnt++] = len;
 98         da(hehe,sa,rank,height,len,300);
 99         //debug(len);
100         LL ans = 0;
101         for(int i=1;i<=len;i++)
102         {
103             int pos = ch[std::lower_bound(ch,ch+cnt,sa[i]) - ch] - sa[i];
104             ans += len -sa[i]  ;
105             ans -= std::max(pos,height[i]);
106         }
107         printf("Case #%d: %I64d\n",++cas,ans);
108     }
109 }