1. 程式人生 > >計蒜客-2018南京網絡賽skr(回文樹)

計蒜客-2018南京網絡賽skr(回文樹)

names 字符集 void space 一個數 out 當前 sum typedef

題意:給你一個數字字符串,問你本質不同的回文子串的和是多少

解題思路:用回文樹直接可以把本質不同的回文子串都跑出來,然後分別遍歷以奇樹為節點和以偶數為節點的兩顆回文樹就行了

代碼:

#include<bits/stdc++.h>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long ll;
typedef long long LL;
const int maxn = 2000000 + 10;
const
int N = 10 ; const int mod=1e9+7; ll sum[maxn]; struct Palindromic_Tree { int next[maxn][N] ;//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 ;//節點指針 LL val[maxn]; int newnode ( int l ) {//新建節點 for ( int i = 0 ; i < 10 ; ++ i ) next[tot][i] = 0 ; cnt[tot] = 0 ; num[tot] = 0 ; val[tot] = 0LL; len[tot] = l ; return tot ++ ; }
void init () {//初始化 tot = 0 ; newnode ( 0 ) ; newnode ( -1 ) ; last = 0 ; n = 0 ; S[n] = -1 ;//開頭放一個字符集中沒有的字符,減少特判 fail[0] = 1 ; } int get_fail ( int x ) //get_fail函數就是讓找到第一個使得S[n - len[last] - 1] == S[n]的last {//和KMP一樣,失配後找一個盡量最長的 while ( S[n - len[x] - 1] != S[n] ) x = fail[x] ;//如果沒有構成回文,那麽去找最長的後綴回文子串 return x ;//如果能構成回文,說明可以通過之前的點+一個字符構成新的回文 } void add ( int c ) { c -= 0 ; S[++ n] = c ; int cur = get_fail ( last ) ;//通過上一個回文串找這個回文串的匹配位置 if ( !next[cur][c] ) {//如果這個回文串沒有出現過,說明出現了一個新的本質不同的回文串 int now = newnode ( len[cur] + 2 ) ;//新建節點 fail[now] = next[get_fail ( fail[cur] )][c] ;//和AC自動機一樣建立fail指針,以便失配後跳轉 next[cur][c] = now ; num[now] = num[fail[now]] + 1 ; } int pre = cur; last = next[cur][c]; cnt[last] ++ ; } void count () { for ( int i = tot - 1 ; i >= 0 ; -- i ) cnt[fail[i]] += cnt[i] ; //父親累加兒子的cnt,因為如果fail[v]=u,則u一定是v的子回文串! } }run; char s[maxn]; ll ans1,ans2; void dfs(int x,ll pre,ll len) { for(ll i=0;i<10;i++) { if(run.next[x][i]) { ll tmp=((sum[len-1]*i)%mod+pre*10+i)%mod;tmp%=mod; ans1+=tmp;ans1%=mod; dfs(run.next[x][i],tmp,len+2); } } } void dfs2(int x,ll pre,ll len) { ll tmp; for(ll i=0;i<10;i++) { if(run.next[x][i]) { if(len==1) { ans2=ans2+i;tmp=i; } else { tmp=((sum[len]*i)%mod+pre*10+i)%mod;tmp%=mod; ans2+=tmp;ans2%=mod; } dfs(run.next[x][i],tmp,len+2); } } } int main() { sum[0]=1; for(int i=1;i<=maxn;i++) { sum[i]=sum[i-1]*10;sum[i]%=mod; } scanf("%s",&s); run.init(); int nn=strlen(s)-1; for(int i=0;i<=nn;i++) run.add(s[i]); dfs(0,0,2); dfs2(1,0,1); ll ans=ans1+ans2; cout<<ans<<endl; }

計蒜客-2018南京網絡賽skr(回文樹)