【浮*光】 #Trie+Hash# 洛谷p3449 [POI2006]PAL-Palindromes

题目描述

给出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 我就是个坑不用在意

 

                                            ——时间划过风的轨迹，那个少年，还在等你。