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+100;
#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*3],wb[maxn*3],wv[maxn*3],wss[maxn*3];

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

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

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

int main()
{
    //freopen("input.txt","r",stdin);
    scanf("%d ",&T);
    int cas = 0;
    while(T--)
    {
        scanf(" %c %s",&x,line);
        int len = strlen(line);
        for(int i=0;i<len;i++) hehe[i] = line[i];
        int ch[maxn],cnt = 0;
        for(int i=0;i<len;i++)
        {
            if(line[i] == x) ch[cnt++] = i;
        }
        ch[cnt++] = len;
        da(hehe,sa,rank,height,len,300);
        //debug(len);
        LL ans = 0;
        for(int i=1;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
",++cas,ans);
    }
}