HDU 5421 Victor and String

Victor and String

Time Limit: 1000ms

Memory Limit: 262144KB

This problem will be judged on HDU. Original ID: 5421
64-bit integer IO format: %I64d      Java class name: Main

Victor loves to play with string. He thinks a string is charming as the string is a palindromic string.

Victor wants to play n times. Each time he will do one of following four operations.

Operation 1 : add a char c to the beginning of the string.

Operation 2 : add a char c to the end of the string.

Operation 3 : ask the number of different charming substrings.

Operation 4 : ask the number of charming substrings, the same substrings which starts in different location has to be counted.

At the beginning, Victor has an empty string.

Input
The input contains several test cases, at most 5 cases.

In each case, the first line has one integer n means the number of operations.

The first number of next n line is the integer op, meaning the type of operation. If op=1 or 2, there will be a lowercase English letters followed.

1≤n≤100000.

Output
For each query operation(operation 3 or 4), print the correct answer.

Sample Input

6
1 a
1 b
2 a
2 c
3
4
8
1 a
2 a
2 a
1 a
3
1 b
3
4

Sample Output

4
5
4
5
11

Source

BestCoder Round #52 (div.1) ($)

 

解题：Palindromic Tree

#include <bits/stdc++.h>
using namespace std;
using LL = long long;
const int maxn = 200010;
struct PalindromicTree {
    int son[maxn][26],fail[maxn],num[maxn],s[maxn],len[maxn];
    int tot,L,R,last[2];
    LL ret;
    int newnode(int slen = 0) {
        num[tot] = 0;
        memset(son[tot],0,sizeof son[tot]);
        len[tot] = slen;
        return tot++;
    }
    void init() {
        ret = tot = last[0] = last[1] = 0;
        L = maxn>>1;
        R = L - 1;
        newnode(0);
        newnode(-1);
        fail[0] = fail[1] = 1;
        memset(s,-1,sizeof s);
    }
    int getFail(int kd,int x) {
        if(kd) while(s[R - len[x] -1] != s[R]) x = fail[x];
        else while(s[L + len[x] + 1] != s[L]) x = fail[x];
        return x;
    }
    void extend(int kd,int c) {
        if(kd) s[++R] = c;
        else s[--L] = c;
        int cur = getFail(kd,last[kd]);
        if(!son[cur][c]) {
            int x = newnode(len[cur] + 2);
            fail[x] = son[getFail(kd,fail[cur])][c];
            son[cur][c] = x;
            num[x] = num[fail[x]] + 1;
        }
        last[kd] = son[cur][c];
        if(len[last[kd]] == R - L + 1) last[kd^1] = last[kd];
        ret += num[last[kd]];
    }
} pt;
int main() {
    int n,op;
    char str[10];
    while(~scanf("%d",&n)) {
        pt.init();
        while(n--) {
            scanf("%d",&op);
            if(op <= 2) {
                scanf("%s",str);
                pt.extend(op-1,str[0] - 'a');
            } else if(op == 3) printf("%d
",pt.tot - 2);
            else if(op == 4) printf("%I64d
",pt.ret);
        }
    }
    return 0;
}