ACM-ICPC 2018 南京赛区网络预赛 I Skr (马拉车+hash去重)或(回文树)

https://nanti.jisuanke.com/t/30998

题意

给一串由0..9组成的数字字符串，求所有不同回文串的权值和。比如说“1121”这个串中有“1”，“2”，“11”，“121”三种回文串，他们的权值分别是1，2，11，121。最终输出ans=135。

分析

第一次知道马拉车是manacher。。。涨姿势了

在马拉车进行的过程中，进行子回文串的统计去重。

这里的哈希去重方法重点学习理解。

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <string>
#include <algorithm>
#include <cmath>
#include <ctime>
#include <vector>
#include <queue>
#include <map>
#include <stack>
#include <set>
#include <bitset>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
#define ms(a, b) memset(a, b, sizeof(a))
#define pb push_back
#define mp make_pair
#define pii pair<int, int>
//#define eps 0.0000000001
#define IOS ios::sync_with_stdio(0);cin.tie(0);
#define random(a, b) rand()*rand()%(b-a+1)+a
#define pi acos(-1)
//const ll INF = 0x3f3f3f3f3f3f3f3fll;
const int inf = 0x3f3f3f3f;
const int maxn = 2e6 + ;
const int maxm =  +;
const int mod = ;

ull base=;
ull p[maxn<<],has[maxn<<];
ll pw[maxn<<],sum[maxn<<];
const int MOD=;
int head[maxn<<],nxt[maxn<<],cnt=;
ull val[maxn];
bool exist(ull now){
    int u=now%MOD;
    for(int i=head[u];i;i=nxt[i]){
        if(val[i]==now) return true;
    }
    val[cnt]=now;
    nxt[cnt]=head[u];
    head[u]=cnt++;
    return false;
}
ull gethas(int l,int r){
    return has[r]-has[l-]*p[r-l+];
}
ll solve(int l,int r){
    ull tmp=gethas(l,r);
    if(exist(tmp)) return ;
    ll ans=(sum[r]-sum[l-]*pw[(r-l++)/]%mod+mod)%mod;
    return ans;
}
char s[maxn];
char Ma[maxn<<];
int Mp[maxn<<];
ll Manacher(char s[],int len){
    int l=;
    Ma[l++]='$';
    Ma[l++]='#';
    for(int i=;i<len;i++){
        Ma[l++]=s[i];
        Ma[l++]='#';
    }
    Ma[l]=;
    pw[]=p[]=;
    has[]=sum[]=;
    for(int i=;i<=l;i++){
        p[i]=p[i-]*base;
        has[i]=has[i-]*base+Ma[i];
        pw[i]=pw[i-]*%mod;
        if(Ma[i]>=''&&Ma[i]<=''){
            sum[i]=(sum[i-]*+Ma[i]-'')%mod;
        }else{
            sum[i]=sum[i-];
        }
    }
    ll ans=;
    int mx=,id=;
    for(int i=;i<l;i++){
        if(Ma[i]!='#') ans=(ans+solve(i,i))%mod;
        Mp[i]=mx>i?min(Mp[*id-i],mx-i):;
        while(Ma[i+Mp[i]]==Ma[i-Mp[i]]){
            if(Ma[i+Mp[i]]!='#') ans=(ans+solve(i-Mp[i],i+Mp[i]))%mod;
            Mp[i]++;
        }
        if(mx<i+Mp[i]){
            mx=i+Mp[i];
            id=i;
        }
    }

    return ans;
}
int main() {
#ifdef LOCAL
    freopen("in.txt", "r", stdin);
//    freopen("input.txt", "w", stdout);
#endif
    scanf("%s",s);
    int len=strlen(s);
    printf("%lld\n",Manacher(s,len));
    return ;
}

回文树的做法是先构建一颗回文树，然后dfs奇偶节点，当前节点的所代表的数字=当前添加的数字*pow(10,当前回文串长度-1) +他父亲的数字*10+当前添加的数字。

比如：33->1331  就是1*1000+33+1。此外有点卡空间，注意内存使用

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <string>
#include <algorithm>
#include <cmath>
#include <ctime>
#include <vector>
#include <queue>
#include <map>
#include <stack>
#include <set>
#include <bitset>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
#define ms(a, b) memset(a, b, sizeof(a))
#define pb push_back
#define mp make_pair
#define pii pair<int, int>
//#define eps 0.0000000001
#define IOS ios::sync_with_stdio(0);cin.tie(0);
#define random(a, b) rand()*rand()%(b-a+1)+a
#define pi acos(-1)
//const ll INF = 0x3f3f3f3f3f3f3f3fll;
const int inf = 0x3f3f3f3f;
const int maxn = 2e6 + ;
const int maxm =  +;
const int mod = 1e9+;

struct PAM{
    int nxt[maxn][];
    int fail[maxn];
    int cnt[maxn];
    int num[maxn];
    int len[maxn];
    int s[maxn];
    int last,n,p;

    int newnode(int w){
        for(int i=;i<;i++) nxt[p][i]=;
        num[p]=cnt[p]=;
        len[p]=w;
        return p++;
    }
    void init(){
        n=last=p=;
        newnode();
        newnode(-);
        s[n]=-;
        fail[]=;
    }
    int get_fail(int x){
        while(s[n-len[x]-]!=s[n]) x=fail[x];
        return x;
    }
    void add(int c){
        c-='';
        s[++n]=c;
        int cur=get_fail(last);
        if(!nxt[cur][c]){
            int now=newnode(len[cur]+);
            fail[now]=nxt[get_fail(fail[cur])][c];
            nxt[cur][c]=now;
            num[now]=num[fail[now]]+;
        }
        last=nxt[cur][c];
        cnt[last]++;
    }
    void Count(){
        for(int i=p-;i>=;i--) cnt[fail[i]]+=cnt[i];
    }
};
PAM pam;
char s[maxn];
ll odd=;
ll even=;
ll qpow(ll a,ll b){
    ll res=;
    while(b){
        if(b&) res=a*res%mod;
        b>>=;
        a=a*a%mod;
    }
    return res;
}
void dfs_odd(int x,ll fa){
    for(int i=;i<=;i++){
        if(pam.nxt[x][i]){
            ll cur;
            if(pam.len[pam.nxt[x][i]]==){
                odd=(i+odd)%mod;
                cur=i;
            }else{
                cur=(i*qpow(,pam.len[pam.nxt[x][i]]-)%mod+i+fa*%mod)%mod;
                odd=(odd+cur%mod)%mod;
            }
            dfs_odd(pam.nxt[x][i],cur);
        }
    }
}
void dfs_even(int x,ll fa){
    for(int i=;i<=;i++){
        if(pam.nxt[x][i]){
            ll cur=(i*qpow(,pam.len[pam.nxt[x][i]]-)%mod+i+fa*%mod)%mod;
            even=(even+cur)%mod;
            dfs_even(pam.nxt[x][i],cur);
        }
    }
}
int main() {
#ifdef LOCAL
    freopen("in.txt", "r", stdin);
//    freopen("input.txt", "w", stdout);
#endif
    pam.init();
    scanf("%s",s);
    int len=strlen(s);
    for(int i=;i<len;i++) pam.add(s[i]);
    dfs_odd(,);
    dfs_even(,);
    printf("%lld\n",(odd+even)%mod);

    return ;
}