SPOJ 705 New Distinct Substrings

SPOJ_705

    历经好几天学习倍增算法的折磨，现在终于能够自己把倍增算法的代码敲出来了。

    对于倍增算法的代码的一些解释可以参考我的另一篇文章：http://www.cnblogs.com/staginner/archive/2012/02/02/2335600.html

    这个题目关键在于要认识到任何一个子串都必然是某一个后缀的前缀，对于任意一个后缀i，它能够贡献出的新的子串的数目等于n-i-height[rank[i]]。其中n-i是其所有的子串的数目，其中由于它和排名前一位的字符串有height[rank[i]]个字符的公共前缀，因此其中有height[rank[i]]个子串是和前面重复的，因次新贡献的子串的数目就等于n-i+height[rank[i]]。

#include<stdio.h>
#include<string.h>
#define MAXD 50010
char b[MAXD];
int r[MAXD], sa[MAXD], rank[MAXD], height[MAXD], wa[MAXD], wb[MAXD], ws[MAXD], wv[MAXD];
int cmp(int *p, int x, int y, int l)
{
    return p[x] == p[y] && p[x + l] == p[y + l];
}
void da(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, x[sa[0]] = 0, p = 1, i = 1; i < n; i ++)
            x[sa[i]] = cmp(y, sa[i - 1], sa[i], j) ? p - 1 : p ++;
    }
}
void calheight(int n)
{
    int i, j, k = 0;
    for(i = 1; i <= n; i ++)
        rank[sa[i]] = i;
    for(i = 0; i < n; height[rank[i ++]] = k)
        for(k ? -- k : 0, j = sa[rank[i] - 1]; r[i + k] == r[j + k]; k ++);
}
void solve()
{
    int i, j, k, n;
    long long int ans;
    gets(b);
    for(i = 0; b[i]; i ++)
        r[i] = b[i];
    r[n = i] = 0;
    da(n + 1, 128);
    calheight(n);
    ans = 0;
    for(i = 0; i < n; i ++)
        ans += n - i - height[rank[i]];
    printf("%lld\n", ans);
}
int main()
{
    int t;
    gets(b);
    sscanf(b, "%d", &t);
    while(t --)
    {
        solve();
    }
    return 0;
}