【HDU4416】Good Article Good sentence【后缀数组】

研究了一会才搞懂。

参照blog（http://blog.csdn.net/acm_cxlove/article/details/7854526 by---cxlove）写的代码。。

先把A和所有的B中间用间隔符串起来，然后求sa，rank，height。

接下来的大体过程是：

求出A与B重复的子串个数 和 A自身重复的子串个数

用A的全部子串个数减去上面求出的值

得到答案。

我们具体来说，比如下面这个例子：

2

aabaabacab

ab

ba

得出排序后的后缀表：

（“{”和“}”都是间隔符，“$”是结尾符，工具来自http://visualgo.net/suffixarray.html）

其中A的长度len为10，即sa值小于10的都是A中的后缀。

我们计算出 对于每个A的后缀，求出与它最近的B的后缀的LCP的长度（“最近”指的是字典序最近），但在实际代码实现中，我们是 先从前往后扫一遍，再从后往前扫一遍更新答案。

具体举例。首先从前往后扫，我们发现第2、3、4、5、6项都是A中的后缀，但第1项不是。这些A中的后缀与第1项B中的后缀的LCP的长度为1，说明这几个A中的后缀每个都有个长为1的子串与B中的后缀重复，我们对于每个A中的后缀记录下这个值。然后从后往前扫，我们发现第6、5、4、3、2项都是A中的后缀，但第7项不是。第4、5、6项与第7项的LCP的长度为2，第2、3项与第7项的LCP的长度为1，说明第4、5、6项有两个子串与B中后缀重复（“a”和“ab”，等等，为什么没“b”？因为“b”会在第10、11、12项被计算，如果这里计算了，那么就会重复计算，所以只考虑从当前后缀开头的子串就不会重复计算），第2、3项有一个子串与B中后缀重复。

注意到从后往前扫时，第4、5、6项的答案更新了（从1变为2，因为多了子串“ab”），所以我们记录每个A中后缀的贡献时，是取的最大值。

完了？当然没有，还有A中自身重复的子串。

对于第2、3项举例说明。我们发现这两个后缀都是属于A的，他们的LCP的长度为4，是大于我们之前得到的2的，说明在A中还有子串“aab”和“aaba”自身重复，但与B没有重复。这也是对于答案有贡献的，我们还是取最大值。于是第二项对于答案的贡献为4。

那么第3项的答案也是4吗？不是，我们考虑的是当前后缀与下一个后缀（当然首先得保证两个后缀都是A里的），即第3项与第4项的LCP的长度，贡献为1。

那么我们用len * (len + 1) / 2减去记录下来的每个值就可以啦。

具体看代码吧。

注意开LL。

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

typedef long long LL;

const int maxn = 300005, inf = 0x3f3f3f3f;

int sa[maxn], rk[maxn], height[maxn];

int wa[maxn], wb[maxn], wv[maxn], cnt[maxn];
void SA(int *r, int n, int m) {
    int *x = wa, *y = wb;
    for(int i = 0; i < m; i++) cnt[i] = 0;
    for(int i = 0; i < n; i++) cnt[x[i] = r[i]]++;
    for(int i = 1; i < m; i++) cnt[i] += cnt[i - 1];
    for(int i = n - 1; i >= 0; i--) sa[--cnt[x[i]]] = i;
    for(int j = 1; j < n; j <<= 1) {
        int p = 0;
        for(int i = n - j; i < n; i++) y[p++] = i;
        for(int i = 0; i < n; i++) if(sa[i] >= j) y[p++] = sa[i] - j;
        for(int i = 0; i < n; i++) wv[i] = x[y[i]];
        for(int i = 0; i < m; i++) cnt[i] = 0;
        for(int i = 0; i < n; i++) cnt[wv[i]]++;
        for(int i = 1; i < m; i++) cnt[i] += cnt[i - 1];
        for(int i = n - 1; i >= 0; i--) sa[--cnt[wv[i]]] = y[i];
        swap(x, y);
        p = 1; x[sa[0]] = 0;
        for(int i = 1; i < n; i++)
            x[sa[i]] = y[sa[i - 1]] == y[sa[i]] && y[sa[i - 1] + j] == y[sa[i] + j] ? p - 1 : p++;
        if(p >= n) break;
        m = p;
    }
}

void calcHeight(int *r, int n) {
    int i, j, k;
    for(i = j = k = 0; i < n; height[rk[i++]] = k)
        for(k ? k-- : 0, j = sa[rk[i] - 1]; r[i + k] == r[j + k]; k++);
}

int n, size, num[maxn], pos[maxn];
char str[maxn];

int main() {
    int T; scanf("%d", &T);

    for(int cas = 1; cas <= T; cas++) {
        for(int i = 0; i < maxn; i++) pos[i] = 0; size = 0; 

        scanf("%d", &n);
        
        scanf("%s", str);
        int len = strlen(str), end = 27;
        for(int i = 0; i < len; i++) num[size++] = str[i] - 'a' + 1;
        for(int i = 1; i <= n; i++) {
            num[size++] = end++;
            scanf("%s", str); int slen = strlen(str);
            for(int j = 0; j < slen; j++) num[size++] = str[j] - 'a' + 1;
        }
        num[size] = 0;

        SA(num, size + 1, end);
        for(int i = 1; i <= size; i++) rk[sa[i]] = i;
        calcHeight(num, size);

        int tmp;
        tmp = inf;
        for(int i = 0; i <= size; i++)
            if(sa[i] < len) {
                tmp = min(tmp, height[i]);
                pos[sa[i]] = max(pos[sa[i]], tmp);
            } else tmp = inf;

        tmp = inf;
        for(int i = size - 1; i >= 0; i--)
            if(sa[i] < len) {
                tmp = min(tmp, height[i + 1]);
                pos[sa[i]] = max(pos[sa[i]], tmp);
            } else tmp = inf;
        
        for(int i = 0; i < size; i++) if(sa[i] < len && sa[i + 1] < len)
            pos[sa[i]] = max(pos[sa[i]], height[i + 1]);

        LL ans = (LL)len * (len + 1) >> 1;
        for(int i = 0; i < len; i++) ans -= pos[i];
        
        printf("Case %d: %I64d\n", cas, ans);
    }

    return 0;
}