题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=6599
解题心得:
- 其实看到这个回文串计数问题就该想到回文树,用回文树得到每一个相同种类回文串的数量,在 h a s h hash hash或者 m a n a c h manach manach判断一下是否前一半也是回文串就行了。
- 比赛的时候这么多人都会回文树也tql吧,都是些什么神仙啊,气馁。
#include <bits/stdc++.h>
using namespace std;
const int maxn = 6e5+100;
char ch[maxn];
int ans[maxn];
struct Palindromic_Tree {
int next[maxn][30] ;//next指针,next指针和字典树类似,指向的串为当前串两端加上同一个字符构成
int fail[maxn] ;//fail指针,失配后跳转到fail指针指向的节点
int cnt[maxn] ;
int num[maxn] ;
int len[maxn] ;//len[i]表示节点i表示的回文串的长度
int s[maxn] ;//存放添加的字符
int last ;//指向上一个字符所在的节点,方便下一次add
int n ;//字符数组指针
int tot ;//节点指针
int endpoint[maxn];
int newnode(int Len) {
memset(next[tot], 0, sizeof(next[tot]));
cnt[tot] = 0;
len[tot] = Len;
num[tot] = 0;
return tot++;
}
void init() {
tot = 0;
newnode(0);
newnode(-1);
last = 0,
n = 0;
fail[0] = 1, fail[1] = 0;
s[n] = -1;
}
int get_fail(int p) {
while(s[n - len[p] -1] != s[n]) p = fail[p];
return p;
}
void insert(int c, int pos) {
s[++n] = c;
int cur = get_fail(last);
if(next[cur][c] == 0) {
int now = newnode(len[cur] + 2);
fail[now] = next[get_fail(fail[cur])][c];
next[cur][c] = now;
num[now] = num[fail[now]] + 1;
}
last = next[cur][c];
cnt[last]++;
endpoint[last] = pos;
}
void add_cnt() {
for(int i=tot-1;i>0;i--) {
cnt[fail[i]] += cnt[i];
}
}
};
struct Hash {
unsigned long long p[maxn], hash[maxn], base = 233;
void init(char s[], int k) {
p[0] = 1;
hash[0] = 0;
for(int i=1;i<=k;i++) p[i] = p[i-1] * base;
for(int i=1;i<=k;i++) hash[i] = hash[i-1]*base + s[i] - 'a' + 1;
}
unsigned long long equal(int l,int r) {
return hash[r] - hash[l-1] * p[r-l+1];
}
};
Palindromic_Tree pam;
Hash hash1, hash2;
bool checke(int pos, int len, int Len) {
int r = pos, l = pos-len+1;
int r1 = Len-l+1, l1 = Len-r+1;
return hash1.equal(l, r) == hash2.equal(l1, r1);
}
int main() {
// freopen("1.in.txt", "r", stdin);
while(scanf("%s", ch+1) != EOF ) {
pam.init();
memset(ans, 0, sizeof ans);
int len = strlen(ch+1);
for(int i=1;i<=len;i++) {
ans[i] = 0;
pam.insert(ch[i]-'a', i);
}
pam.add_cnt();
hash1.init(ch, len);
reverse(ch+1, ch+1+len);
hash2.init(ch, len);
for(int i=2;i<pam.tot;i++) {
int cntlen = pam.len[i];
int pos = pam.endpoint[i];
if(checke(pos - cntlen/2, (cntlen+1)/2, len)) {
ans[cntlen] += pam.cnt[i];
}
}
for(int i=1;i<=len;i++) {
printf("%d", ans[i]);
if(i != len) printf(" ");
}
printf("\n");
}
return 0;
}