ECNUOJ 2619 询问

询问

Time Limit:2000MS Memory Limit:65536KB
Total Submit:286 Accepted:70

Description 

Pollux最近对字符串匹配很感兴趣，他渐渐发现一个有趣的函数，并且他觉得利用这个函数或许可以建立一种新的匹配算法：
对于字符串S[1…n]和i∈[1 , n] ，定义F(i) 为S 和 S[1…i] 的最长公共后缀的长度.
例如对于串S[1…11] = zaaxbaacbaa ，则F(1) = 0 ，F(2) = 1 ， F(3) = 2( 注意S[1…3] = zaa ) ，F(4) = 0 ， …… F(10) = 1 ，F(11) = 11 ；
对于串S[1…n]，i∈[1 , n] ，S[i…n] 为S的后缀；
但是有一点让他犯难的地方就是他不知道如何快速的计算这个函数在i∈[1 , n]的值，作为ECNU的一名编程高手，你能帮助他解决这个问题吗？

Input 

第一行为一个整数T,表示测数数据的组数.
对于每组数据：
第一行为一个字符串S，只由小写字母组成，如果len(s)表示s的长度，那么1 <= len(s) <= 1000000 ；
接下来一个数N( 1 <= N <= 100000 )，表示询问的次数；
接下来N行，每行一个数x( 1<=x<=len(s))。

Output 

对于每个x输出F(x);

Sample Input 

1
zaaxbaacbaa
11
1
2
3
4
5
6
7
8
9
10
11

Sample Output 

0
1
2
0
0
1
3
0
0
1
11

Source

 

解题：没想到别的什么办法，只好用后缀数组了

#include <bits/stdc++.h>
using namespace std;
const int maxn = 100010;
char s[maxn];
int sa[maxn],t[maxn],t2[maxn];
int height[maxn],rk[maxn],c[maxn],n;
void build_sa(int m) {
    int i,*x = t,*y = t2;
    for(i = 0; i < m; ++i) c[i] = 0;
    for(i = 0; i < n; ++i) c[x[i] = s[i]]++;
    for(i = 1; i < m; ++i) c[i] += c[i-1];
    for(i = n-1; i >= 0; --i) sa[--c[x[i]]] = i;

    for(int k = 1; k <= n; k <<= 1) {
        int p = 0;
        for(i = n-k; i < n; ++i) y[p++] = i;
        for(i = 0; i < n; ++i)
            if(sa[i] >= k) y[p++] = sa[i] - k;
        for(i = 0; i < m; ++i) c[i] = 0;
        for(i = 0; i < n; ++i) c[x[y[i]]]++;
        for(i = 1; i < m; ++i) c[i] += c[i-1];
        for(i = n-1; i >= 0; --i) sa[--c[x[y[i]]]] = y[i];
        swap(x,y);
        x[sa[0]] = 0;
        for(p = i = 1; i < n; ++i)
            x[sa[i]] = y[sa[i]] == y[sa[i-1]] && y[sa[i]+k] == y[sa[i-1]+k]?p-1:p++;
        if(p >= n) break;
        m = p;
    }
}
void getHeight() {
    int i,j,k = 0;
    for(i = 0; i < n; ++i) rk[sa[i]] = i;
    for(i = 0; i < n; ++i) {
        if(k) --k;
        j = sa[rk[i]-1];
        while(i + k < n && j + k < n && s[i+k] == s[j+k]) ++k;
        height[rk[i]] = k;
    }
}
int st[maxn][20],m;
void initRMQ() {
    memset(st,0,sizeof st);
    for(int i = 1; i < n; ++i) st[i][0] = height[i];
    for(int i = 1; (1<<i) < n; ++i)
        for(int j = 1; j + (1<<i) <= n; ++j)
            st[j][i] = min(st[j][i-1],st[j+(1<<(i-1))][i-1]);
}
int query(int s,int t) {
    if(s == t) return n - 1 - s;
    s = rk[s];
    t = rk[t];
    if(s > t) swap(s,t);
    s++;
    int k = log2(t - s + 1);
    return min(st[s][k],st[t-(1<<k)+1][k]);
}
int main() {
    int kase;
    scanf("%d",&kase);
    while(kase--) {
        scanf("%s",s);
        n = strlen(s)+1;
        reverse(s,s+n-1);
        build_sa(256);
        getHeight();
        initRMQ();
        scanf("%d",&m);
        while(m--) {
            int tmp;
            scanf("%d",&tmp);
            printf("%d\n",query(0,n - 1 - tmp));
        }
    }
    return 0;
}