1. 程式人生 > >計蒜客 2018南京網路賽 I Skr(迴文樹)

計蒜客 2018南京網路賽 I Skr(迴文樹)

題目:給一串由0..9組成的數字字串,求所有不同迴文串的權值和。比如說“1121”這個串中有“1”,“2”,“11”,“121”三種迴文串,他們的權值分別是1,2,11,121。最終輸出ans=135。

思路:之前寫了馬拉車的演算法,網上看到的這個題是迴文樹的模板題。。。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAXN=2000000+10;
const ll mod=1e9+7;
const int N = 10 ;
ll qmod(ll x,ll p)
{
    ll ans=1;
    while(p)
    {
        if(p&1) ans=ans*x%mod;
        x=x*x%mod;
        p>>=1;
    }
    return ans;
}
struct Palindromic_Tree {
    int next[MAXN][N] ;
    int fail[MAXN] ;
    int cnt[MAXN] ;
    int num[MAXN] ;
    int len[MAXN] ;
    int S[MAXN] ;
    int last ;
    int n ;
    int p ;
    ll sum[MAXN];

    int newnode ( int l ) {
        for ( int i = 0 ; i < N ; ++ i ) next[p][i] = 0 ;
        cnt[p] = 0 ;
        num[p] = 0 ;
        sum[p]=0;
        len[p] = l ;
        return p ++ ;
    }

    void init () {
        p = 0 ;
        newnode (  0 ) ;
        newnode ( -1 ) ;
        last = 0 ;
        n = 0 ;
        S[n] = -1 ;
        fail[0] = 1 ;
    }

    int get_fail ( int x ) {
        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] ;
            next[cur][c] = now ;
            num[now] = num[fail[now]] + 1 ;

            sum[now] = ((sum[cur]*10ll)%mod + c)%mod;
            if(len[cur]>=0)
                sum[now] = (sum[now] + (c*qmod(10,len[cur]+1))%mod)%mod;

        }
        last = next[cur][c] ;
        cnt[last] ++ ;
    }

    ll Sum () {
        ll ans=0;
        for ( int i = p - 1 ; i >= 0 ; -- i )
            (ans += sum[i])%=mod;
        return ans;
    }
}pam;

char s[MAXN];
int main()
{
    scanf("%s",s+1);
    int l=strlen(s+1);
    pam.init();
    for(int i=1;i<=l;i++)
        pam.add(s[i]);
    printf("%lld\n",pam.Sum());
    return 0;
}