题目描述
给出n个回文串s1, s2, …, sn,求如下二元组(i, j)的个数:si+sj仍然是回文串。
简要分析
注意n个串是回文串的条件,这样会使倒序hash方便很多。
所以直接选择字典序hash。把所有串插入一个trie树,记录每个串的hash值。
枚举每个串的每个前缀,如果能和其他串的结束点重合,就判断正反hash。
正反hash匹配的方式为:
代码实现
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#include<vector>
#include<deque>
#include<cmath>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
/*【p3449】PAL
给出n个回文串s1, s2, …, sn,求如下二元组(i, j)的个数:si+sj仍然是回文串。*/
/*【分析】字典序hash。
把所有串插入一个trie树,然后统计每个串的hash,然后再枚举每个串,沿trie树向下走,
枚举每一个前缀,判断他们俩连起来的字符串正着和反着是否一样,直接hash判断即可。*/
#define base 1331
int n,tot=1;
int len[2000005],cnt[2000005],tail[2000005],trie[2000005][26];
//技巧:多种查询时,将trie树的标记数组tail设为某int值,即:对于某些串才为true
ull ans=0,hashs[2000005],powers[2000005];
char ch[2000005];
string ss[2000005];
int main(){
scanf("%d",&n); powers[0]=1; //↓↓预处理power数组
for(int i=1;i<=2000000;i++) powers[i]=powers[i-1]*base;
for(int i=1;i<=n;i++){
scanf("%d",&len[i]);
scanf("%s",ch); ss[i]=ch; //用char读入,再转成string
int p=0; ull tmp=0;
for(int j=0;j<len[i];j++){
int ch=ss[i][j]-'a';
if(!trie[p][ch]) trie[p][ch]=++tot;
p=trie[p][ch]; tmp=tmp*base+(ull)(ch+1);
} cnt[p]++; tail[p]=i; hashs[i]=tmp; //存下每个单词的hash值
} for(int i=1;i<=n;i++){ //枚举前缀
int p=0; for(int j=0;j<len[i];j++){
p=trie[p][ss[i][j]-'a']; //进入前缀的下一个位置,如果有串在此结尾
if(cnt[p]&&hashs[tail[p]]*powers[len[i]]+hashs[i]==
hashs[i]*powers[len[tail[p]]]+hashs[tail[p]]) ans+=cnt[p];
} //↑↑判断二者相连的字符串正着和反着是否一样(因为是回文,所以直接正反判断hash值即可)
} ans=ans*2-n; printf("%lld\n",ans);
}
p.s.蒟蒻第一次尝试绘图qaq 我就是个坑不用在意
——时间划过风的轨迹,那个少年,还在等你。