ZOJ 3494 BCD Code 数位DP+AC自动机

最新推荐文章于 2024-01-20 16:44:22 发布

little_胖

最新推荐文章于 2024-01-20 16:44:22 发布

阅读量576

点赞数

CC 4.0 BY-SA版权

分类专栏： AC自动机 DP 文章标签：算法

本文链接：https://blog.youkuaiyun.com/wangjie_wang/article/details/10083499

DP 同时被 2 个专栏收录

66 篇文章

订阅专栏

AC自动机

2 篇文章

订阅专栏

本文详细介绍了如何使用数位DP和AC自动机解决字符串匹配问题，通过预先处理字符串并利用数位DP进行状态转移，有效提高了算法效率。经过长时间的调试，最终成功解决了挑战性的问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

其实还是比较裸的数位DP，只不过需要用AC自动机来预先处理一下，写了3个小时，各种调试，蛋碎了一地啊。不过幸好最后还是做出来了。

思路：先用AC自动机预处理字符串，数位DP时以dp[index][pos][fg]来表示状态，分别为层数，判断到AC自动机的位置，以及是否前面出现过非0位。

代码略搓：

#include <cstdio>
#include <cstring>
#include <string>
#include <iostream>
#include <map>
#include <set>
#include <vector>
#include <cmath>
#include <stack>
#include <queue>
#include <cstdlib>
#include <algorithm>
using namespace std;
//typedef __int64 int64;
typedef long long ll;
#define M 500005
#define N 1000005
#define max_inf 0x7f7f7f7f
#define min_inf 0x80808080
#define mod 1000000009

struct node//树节点
{
	node *fail , *next[2];
	int flag , pos;
	node()
	{
		fail = NULL;
		memset(next , NULL , sizeof next);
		flag = 0;
	}
}*q[2005] , *rt;
int n , dig[1005] , dp[805][2005][2] , cnt , table[1005];
char A[205] , B[205];

void Insert(char *str)//构造树
{
	int i = 0;
	node *p = rt;
	while (str[i])
	{
		int index = str[i]-'0';
		if (!p->next[index])
		{
			p->next[index] = new node;
			p->next[index]->pos = cnt++;//记录每个点的位置
		}
		p = p->next[index];
		i++;
	}
	p->flag = 1;//标记为一个字符串的末尾
}

void Build()//广搜建立失败指针
{
	int i , head , tail;
	rt->fail = NULL;
	head = tail = 0;
	q[tail++] = rt;
	while (head < tail)
	{
		node *p = q[head++];
		for (i = 0 ; i < 2 ; i++)
		{
			if (!p->next[i])continue;
			if (p == rt)p->next[i]->fail = rt;
			else
			{
				node *temp = p->fail;
				while (temp)
				{
					if (temp->next[i])
					{
						p->next[i]->fail = temp->next[i];
						break;
					}
					temp = temp->fail;
				}
				if (!temp)p->next[i]->fail = rt;
			}
			q[tail++] = p->next[i];
		}
	}
}

int Dfs(int index , node *p , int pos, int fg , int lim)
{
	if (!index)return fg;
	if (!lim && dp[index][pos][fg] != -1)return dp[index][pos][fg];
	int i , j , up = lim ? dig[index] : 9;
	int ret = 0;
	for (i = 0 ; i <= up ; i++)
	{ 
		if (i == 0 && !fg)
		{
			ret += Dfs(index-1 , p , pos , fg||i!=0 , lim&&i==up);
			ret %= mod;
			continue;
		}
		
		//判断这个数字是否能与禁止字符匹配
		int t = i , bit[4];
		for (j = 3 ; j >= 0 ; j--)
		{
			bit[j] = t&1;
			t >>= 1;
		}
		int f = 0;
		node *temp = p;
		for (j = 0 ; j <= 3 ; j++)
		{
			while (temp != rt && !temp->next[bit[j]])temp = temp->fail;
			temp = temp->next[bit[j]];
			if (!temp)temp = rt;
			node *q = temp;
			while (q != rt)
			{
				if (q->flag)f = 1;
				q = q->fail;
			}
		}

		if (!f)//若不匹配则继续往下搜
		{
			ret += Dfs(index-1 , temp , temp->pos , fg||i!=0 , lim&&i==up);
			ret %= mod;
		}
	}
	if (!lim)dp[index][pos][fg] = ret;
	return ret;
}

int Solve(char *num)
{
	int len = 0;
	int i;
	for (i = strlen(num)-1 ; i >= 0 ; i--)dig[++len] = num[i]-'0';
	return Dfs(len , rt , rt->pos , 0 , 1);
}

int Judge()//单独判断数字A是否满足条件
{
	int i = 0 , j;
	node *p = rt;
	while (A[i])
	{
		int t = A[i]-'0' , bit[4];
		for (j = 3 ; j >= 0 ; j--)
		{
			bit[j] = t&1;
			t >>= 1;
		}
		for (j = 0 ; j <= 3 ; j++)
		{
			while (p != rt && !p->next[bit[j]])p = p->fail;
			p = p->next[bit[j]];
			if (!p)p = rt;
			node *q = p;
			while (q != rt)
			{
				if (q->flag)return 0;
				q = q->fail;
			}
		}
		i++;
	}
	return 1;
}

int main()
{
	int t;
	scanf("%d",&t);
	while (t--)
	{
		cnt = 0;
		rt = new node;
		rt->pos = cnt++;
		memset(dp , -1 , sizeof dp);
		scanf("%d",&n);
		while (n--)
		{
			char str[25];
			scanf("%s",str);
			Insert(str);
		}
		Build();
		scanf("%s%s",A,B);
		printf("%d\n",(Solve(B)-Solve(A)+Judge()+mod)%mod);
	}
	return 0;
}