HDU 6086 Rikka with String(2017 Multi-University Training Contest 3)

题目链接Rikka with String
题意:给定 n 01字符串和一个长度 L ,问所有长度为2L的反对称字符串( s[i]s[|s|i+1],i[1,|s|] )中出现所有给定字符串的种类数。
题解:我们先考虑一个简化版的问题:所有长度为 L 的字符串中出现所有给定字符串的种类数。这就是一个基本的AC自动机上dp的问题,记长度为 i ,当前节点为j且为字符串k的末尾节点,F[x]代表 x 的后继节点集合,那么转移方程为
For every pF[j]

dp[i+1][p][state|(1<<k)]+=dp[i][j][state]

那么对于原问题来说,我们可以分情况讨论:
1 .字符串全部在[1,|S|]中,这部分和简化版问题做法一致。
2 .字符串一部分在[1,|S|]一部分在 [|S|+1,2|S|] 里,这部分可以直接枚举每一个字符串的对称轴,然后对于合法的字符串插入到AC自动机里。
3 .字符串全部在[|S|+1,2|S|]里,我们直接取对应的反字符串,然后就和情况 1 一样了。
我们将上述三种情况分类讨论,然后建立AC自动机,并在上面做上面写的dp即可。

#include <bits/stdc++.h>

using namespace std;

const int N = 245;
const int M = 2;
int id,son[N][M],fail[N],ifo1[N],ifo2[N],q[N];

void Ist(const char* s,int info,bool ps){
    int x=0,y;
    for(int i=0;s[i];i++){
        y=s[i]-'0';
        if(!son[x][y])
            son[x][y]=++id;
        x=son[x][y];
    }
    ps?ifo1[x]|=1<<info:ifo2[x]|=1<<info;
}
void Build(){
    int hd=0,tl=0,x;
    for(int i=0;i<2;i++)
        if(son[0][i])
            q[tl++]=son[0][i];
    while(hd!=tl){
        x=q[hd++];
        for(int i=0;i<2;i++)
            if(son[x][i]){
                q[tl++]=son[x][i];
                fail[son[x][i]]=son[fail[x]][i];
                ifo1[son[x][i]]|=ifo1[son[fail[x]][i]];
                ifo2[son[x][i]]|=ifo2[son[fail[x]][i]];
            }else
                son[x][i]=son[fail[x]][i];
    }
}

const long long MOD = 998244353;
long long dp[2][N][110];
char s[25],t[25];

int main(){
    int T,len;
    scanf("%d",&T);
    while(T--){
        int n,l;
        id=0;
        memset(son,0,sizeof son);
        fill(fail,fail+N,0);
        fill(ifo1,ifo1+N,0);
        fill(ifo2,ifo2+N,0);
        scanf("%d %d",&n,&l);
        for(int i=0;i<n;i++){
            scanf(" %s",s);
            Ist(s,i,true);
            len=strlen(s);
            for(int j=0;j<len;j++){
                bool ck=true;
                for(int k=j;k<min(len,j*2);k++)
                    if(s[k]==s[2*j-k-1]){
                        ck=false;
                        break;
                    }
                if(ck){
                    if(j*2<len){
                        for(int k=len-1;k>=j;k--)
                            t[len-1-k]=(s[k]=='0'?'1':'0');
                        t[len-j]='\0';
                    }else{
                        for(int k=0;k<j;k++)
                            t[k]=s[k];
                        t[j]='\0';
                    }
                    if(!j)
                        Ist(t,i,true);
                    else
                        Ist(t,i,false);
                }
            }
        }
        Build();
        memset(dp,0,sizeof dp);
        dp[0][0][0]=1;
        long long ans=0;
        for(int i=0;i<l;i++)
            for(int j=0;j<=id;j++)
                for(int k=0;k<(1<<n);k++)
                    if(dp[i%2][j][k]){
                        for(int p=0;p<2;p++){
                            int ps=son[j][p];
                            int stt=k|ifo1[ps];
                            if(i+1==l)
                                stt|=ifo2[ps];
                            dp[(i+1)%2][ps][stt]=
                            (dp[(i+1)%2][ps][stt]+dp[i%2][j][k])%MOD;
                        }
                        dp[i%2][j][k]=0;
                    }
        for(int i=0;i<=id;i++)
            ans=(ans+dp[l%2][i][(1<<n)-1])%MOD;
        printf("%lld\n",ans);
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值