BZOJ 2746 旅行问题 （fail树 Hash LCA）

2746: [HEOI2012]旅行问题

Time Limit: 30 Sec Memory Limit: 256 MB
Description

yz是Z国的领导人，他规定每个地区的名字只能为26个小写拉丁字母的一个。由于地 区数有可能超过26个，便产生了一个问题，如何辨别名字相同的地区？于是yz规定，一个 地区的描述必须包含它的所有上级，且上级按次序排列。于是，一个地区的描述是一个字符 串。比如说，一个地区的名字为c，它的上级为b，b的上级为a，a没有上级，那么这个地 区就描述为abc。显然，这个描述同时包含了c的上级b和b的上级a的描述，分别为ab和a。 值得注意的是，每个地区最多有一个上级，同一上级的地区之间名字不同，没有上级的 地区之间名字不同。现在，yz对外公布了n个地区的描述，这些描述中包含了Z国所有地区的描述，并让 你处理来访者的旅行问题。现有m对人访问这个国家，对于每对人，第一个人喜欢第i个描述中的第j个地区，设 这个地区描述为s1，第二个人喜欢第k个描述中的第l个地区，设这个地区描述为s2。他们为了统一行程，决定访问描述为s的地区（显然他们只关心地区的名字，并非是地区本身）， 设s的长度为t，s需要满足以下条件：
1：t<=j, t<=l;
1：s[1..t] = s1[j-t+1 … j], s[1..t] = s2[l-t+1 … l]；（即s为s1中1到k位 与s2中1到l位的公共后缀）
2：t最大化。
为了不使输出过大，你只需把这个字符串按照如下生成的26进制数转成10进制后mod 1000000007后输出:
a->0
b->1
.
.
.
z->25
比如地区cab被编码成2 * 26^2 + 0 * 26^1 + 1 * 26^0 = 1353。

Input

第一行给定一个整数n
第2…n+1行:每i+1行给定一个字符串a[i],表示第i个描述。
接下来一行一个整数m
接下来m行:每行给定四个整数i,j,k,l，字母含义与题目描述一致。

Output

共m行，每行一个整数，表示答案字符串的编码。

Sample Input

2
aabb babb
2
1 3 2 3
1 4 2 4

Sample Output

1
1

【样例说明】

询问1中的公共后缀有ab和b，但是没有ab这个地区，只有b地区，所以只能选择b这个 地区；

询问2中的公共后缀有abb、bb和b，但是没有abb和bb这两个地区，只有b地区，所以 只能选择b这个地区。

HINT

【数据范围】
设这个国家地区总数数为tot（注意：输入的字符串总长度可能超过tot！）
对于30%的数据，满足tot，m，n<=100；
对于50%的数据，满足tot，m，n<=1000；
对于80%的数据，满足tot，m，n<=100000；
对于100%的数据，满足tot，m，n<=1000000；
保证输入文件不超过20MB。

思路：
建出Trie，做个AC自动机，然后查询fail树上的LCA即可。

#include <cstdio>
#include <iostream>
#include <cstring>
using namespace std;

const int N = 1 << 20;
const int mod = 1e9 + 7;

int n, m;
int dep[N], q[N];
int Hash[N];
int fail[N][21];
int next[N][26], tot = 1;
int st[N], pos[N << 1], ord;

inline void insert(char *s){
    int t = 1, idc = -1;//1作为root 
    while( s[++idc] ){
        int c = s[idc] - 'a';
        if(next[t][c] == 0){
            next[t][c] = ++tot;
            Hash[tot] = (Hash[t] * 26LL + c) % mod;
        }
        t = next[t][c];
        pos[++ord] = t;//记录每个地址在fail树上的位置 
    }
}

inline void build(){//建个Trie树，连上fail，处理fail树 
    int l=0, r=0;
    memset(q, 0, sizeof(q));
    fail[1][0] = 1;
    for(int i=0; i<26; ++i)
        if(next[1][i]){
            fail[next[1][i]][0] = 1;
            q[r++] = next[1][i];
        }
    while(l < r){
        int t = q[l++];
        if(t != 1) dep[t] = dep[fail[t][0]] + 1;
        for(int i=1; i<=20; ++i)
            fail[t][i] = fail[fail[t][i-1]][i-1];//st表的方式建fail，相当于预处理了fail树上的st 
        for(int i=0; i<26; ++i){
            if(next[t][i]){
                int p = fail[t][0];
                while(p != 1 && !next[p][i]) p = fail[p][0];//一直向上跳 
                if(next[p][i]) fail[next[t][i]][0] = next[p][i];
                else fail[next[t][i]][0] = 1;
                q[r++] = next[t][i];
            }
        }
    }
}

inline int lca(int a, int b){//从两个原串的pos向上跳，就是一直找是前缀的后缀，第一个公共的就是ans 
    if(dep[a] < dep[b]) swap(a, b);
    for(int i=20; i>=0; --i)
        if(dep[fail[a][i]] >= dep[b])
            a = fail[a][i];
    if(a == b) return a;
    for(int i=20; i>=0; --i)
        if(fail[a][i] != fail[b][i])
            a = fail[a][i], b = fail[b][i];
    return fail[a][0];
}

int main(){
    freopen ("alicemagic.in", "r", stdin);
    freopen ("alicemagic.out", "w", stdout);
    scanf("%d", &n);
    for(int i=1; i<=n; ++i){
        char str[N];
        scanf("%s", str);
        st[i] = ord;//之前有ord个字符 
        insert(str);
    }
    build();
    scanf("%d", &m);
    for (int i = 0; i < m; ++i){
        int s1, s2, l1, l2, p1, p2;
        scanf("%d%d%d%d", &s1, &l1, &s2, &l2);
        p1 = pos[st[s1] + l1];//找到地址在fail树上的pos 
        p2 = pos[st[s2] + l2];
        int t = lca(p1, p2);       
        printf("%d\n", Hash[t]);
    }
}