UVALive 4126 Password Suspects(AC自动机+dp)

题意就是长度为n的串,满足下列m个串是它的子串,问这个串有多少种,如果少于42种则字典序输出。

dp[i][j][k]为,填充了i个字符,在ac自动机的j节点上,字串集合为k的方案数,填充i个字符可以从前往后,或者从后往前填,但是如果要用dfs,最好选用从后往前填,因为如果这样dp[i][j][k]肯定是由j这个节点选择a-z得到的dp转移过来的,但是如果从前往后,那么就要知道那几个节点能转移到j这个节点。这点原因和uvalive 3490是差不多的。

因为递推方向是从后往前,所以在输出字典序的时候,都不需要保存在某个容器里排序什么的,枚举的时候是a-z,那么找到的解直接就是按字典序排序的。如果方向是从前往后,首先终点就有很多个比较麻烦不说,并且无法保证先找到的字典序小。

AC代码:

#pragma comment(linker, "/STACK:102400000,102400000")
#include<cstdio>
#include<ctype.h>
#include<algorithm>
#include<iostream>
#include<cstring>
#include<vector>
#include<cstdlib>
#include<stack>
#include<queue>
#include<set>
#include<map>
#include<cmath>
#include<ctime>
#include<string.h>
#include<string>
#include<sstream>
#include<bitset>
using namespace std;
#define ll long long
#define ull unsigned long long
#define eps 1e-8
#define NMAX 201000
#define MOD 1000000007
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define PI acos(-1)
template<class T>
inline void scan_d(T &ret)
{
    char c;
    int flag = 0;
    ret=0;
    while(((c=getchar())<'0'||c>'9')&&c!='-');
    if(c == '-')
    {
        flag = 1;
        c = getchar();
    }
    while(c>='0'&&c<='9') ret=ret*10+(c-'0'),c=getchar();
    if(flag) ret = -ret;
}
const int maxnode = 11*11+10;

int end[maxnode];
ll dp[26][maxnode][1050];//对于dp[i][j][k],从最后一个往前填,意义是第i个节点是j,包含字符串集合为k的个数。

struct AhoCorasick
{
    int ch[maxnode][26];
    int f[maxnode];
//    int last[maxnode],val[maxnode];
    int sz;
    void init()
    {
        memset(ch[0],0,sizeof(ch[0]));
        sz = 1;
    }
    int idx(char c)
    {
        return c-'a';
    }
    void insert(char *s,int v)
    {
        int u = 0, n = strlen(s);
        for(int i= 0; i < n; i++)
        {
            int c = idx(s[i]);
            if(!ch[u][c])
            {
                memset(ch[sz],0,sizeof(ch[sz]));
                end[sz] = 0;
                ch[u][c] = sz++;
            }
            u = ch[u][c];
        }
        end[u] |= 1<<v;
    }
    void getFail()
    {
        queue<int>q;
        f[0] = 0;
        for(int c = 0; c < 26; c++)
        {
            int u = ch[0][c];
            if(u){f[u] = 0; q.push(u);
//            last[u] = 0;
            }
        }
        while(!q.empty())
        {
            int r = q.front(); q.pop();
            for(int c = 0; c < 26; c++)
            {
                int u = ch[r][c];
                if(!u)
                {
                    ch[r][c] = ch[f[r]][c];
                    continue;
                }
                q.push(u);
                int v = f[r];
                while(v && !ch[v][c]) v = f[v];
                f[u] = ch[v][c];
                end[u] |= end[f[u]];
//                last[u] = val[f[u]] ? f[u] : last[f[u]];
            }
        }
    }
}ac;

char ss[30];

int n,m;
ll dfs(int len, int pos, int s)
{
    ll &ans = dp[len][pos][s];
    if(ans != -1) return ans;
    if(len == n)
    {
        if(s == (1<<m)-1) return ans = 1;
        return ans = 0;
    }
    ans = 0;
    for(int i = 0; i < 26; i++)
    {
        int next = ac.ch[pos][i];
        ans += dfs(len+1, next, s | end[next]);
    }
    return ans;
}

void print(int len, int pos, int s)
{
    if(len == n)
    {
        ss[len] = '\0';
        printf("%s\n",ss);
        return;
    }
    for(int i = 0; i < 26; i++)
    {
        int next = ac.ch[pos][i];
        ss[len] = i+'a';
        if(dp[len+1][next][s|end[next]] > 0) print(len+1, next, s|end[next]);
    }
}

int main()
{
#ifdef GLQ
    freopen("input.txt","r",stdin);
//    freopen("o3.txt","w",stdout);
#endif
    int cas = 1;
    while(~scanf("%d%d",&n,&m) && n+m)
    {
        ac.init();
        for(int i = 0; i < m; i++)
        {
            scanf("%s",ss);
            ac.insert(ss,i);
        }
        ac.getFail();
        memset(dp,-1,sizeof(dp));
        ll ans = dfs(0,0,0);
        printf("Case %d: %lld suspects\n",cas++,ans);
        if(ans <= 42) print(0,0,0);
    }
    return 0;
}


  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值