HDOJ 3948 The Number of Palindromes 回文串自动机

看上去像是回文串自动机的模板题，就来了一发

The Number of Palindromes

Time Limit: 6000/3000 MS (Java/Others)    Memory Limit: 262144/262144 K (Java/Others)

Total Submission(s): 1992    Accepted Submission(s): 694


Problem Description

Now, you are given a string S. We want to know how many distinct substring of S which is palindrome.

 

Input

The first line of the input contains a single integer T(T<=20), which indicates number of test cases.

Each test case consists of a string S, whose length is less than 100000 and only contains lowercase letters.

 

Output

For every test case, you should output "Case #k:" first in a single line, where k indicates the case number and starts at 1. Then output the number of distinct substring of S which is palindrome.

 

Sample Input

3
aaaa
abab
abcd

 

Sample Output

Case #1: 4
Case #2: 4
Case #3: 4

 

Source

2011 Multi-University Training
Contest 11 - Host by UESTC

 

/* ***********************************************
Author        :CKboss
Created Time  :2015年04月17日 星期五 09时38分22秒
File Name     :pt.cpp
************************************************ */

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <string>
#include <cmath>
#include <cstdlib>
#include <vector>
#include <queue>
#include <set>
#include <map>

using namespace std;

const int C=30;
const int maxn=120000;

struct Node
{
Node *ch[C],*suffix;
int len;
}bar[maxn],*foo,*last,*odd,*even;

char s[maxn];
int n,cnt;
// cnt=foo-bar=count of palindromes , n the number of char added

void init()
{
odd=bar; even=last=odd+1; foo=even+1;
memset(odd->ch,0,sizeof(odd->ch));
memset(even->ch,0,sizeof(even->ch));
odd->suffix=even->suffix=odd;
odd->len=-1; even->len=0;
n=0; cnt=0;
}

Node* New_Node(int x)
{
memset(foo->ch,0,sizeof(foo->ch));
foo->len=x;
return foo++;
}

int index(char x) { return x-'a'; }

Node* get(Node*p)
{
while(n-p->len-2<0||s[n-p->len-2]!=s[n-1])
p=p->suffix;
return p;
}

bool add(char c)
{
int x=index(c); s[n++]=c;
Node* p =get(last);
if(!p->ch[x])
{
last = New_Node(p->len+2);
if(last->len==1) last->suffix=even;
else last->suffix=get(p->suffix)->ch[x];
/// guarantee proper suffix
p->ch[x]=last; cnt++;
return true;
}
else
{
last=p->ch[x];
return false;
}
}

char str[maxn];

int main()
{
int T_T,cas=1;
scanf("%d",&T_T);
while(T_T--)
{
scanf("%s",str);
int len=strlen(str);
init();
for(int i=0;i<len;i++)
{
add(str[i]);
}
printf("Case #%d: %d\n",cas++,cnt);
}
}