数位DP

最新推荐文章于 2024-01-26 11:41:42 发布

Love_xyh

最新推荐文章于 2024-01-26 11:41:42 发布

阅读量100

点赞数 1

分类专栏：数位DP

本文链接：https://blog.csdn.net/Dove_xyh/article/details/101917453

版权

数位DP 专栏收录该内容

5 篇文章 0 订阅

订阅专栏

新学数位DP，从目前学到的角度来看，数位DP主要有种类型：

1.让你求l—r区间内符合要求的数的个数

2.让你求l—r区间内符合要求的数的 “数值的和” 或是 “所有位上数字加起来的和”

其实对于后一种类型来说，只要再记忆化搜索过程中多返回某些需要记忆的值即可。

对于类型一，就不过多介绍了，下面推荐几道类型二的好题。

CF1073E Segment Sum

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int N=20,MOD=998244353;
int n,m,k;
int a[N],dp[N][1<<12],f[N][1<<12],inv[N];

struct node{int num,sum;};  //num是个数，sum是和 

node dfs(int x,int pre,int p,int cnt,int lead,int limit)
{
	if (x==0)
	{
		if (lead) return (node){0,0};
		else return (node){1,0};
	}
	if (!lead && !limit && dp[x][p]!=-1)
	{
		return (node){dp[x][p],f[x][p]};
	}
	int h;
	if (limit) h=a[x]; else h=9;
	node res,now;
	res.num=res.sum=now.num=now.sum=0;
	for (register int i=0; i<=h; ++i)
	{
		if (cnt==k && ((1<<i)&p)==0) continue;
		if (lead && (i==0))
		{
			now=dfs(x-1,-10,p,cnt,1,(i==h)&&limit);
			res.num=(res.num+now.num)%MOD;
			res.sum=(res.sum+now.sum)%MOD;
			res.sum=(res.sum+i*inv[x]%MOD*now.num%MOD)%MOD;
			continue;
		}
		if ((1<<i)&p)
		{
			now=dfs(x-1,i,p,cnt,0,(i==h)&&limit);
			res.num=(res.num+now.num)%MOD;
			res.sum=(res.sum+now.sum)%MOD;
			res.sum=(res.sum+i*inv[x]%MOD*now.num%MOD)%MOD;
		}
		else 
		{
			now=dfs(x-1,i,(1<<i)|p,cnt+1,0,(i==h)&&limit);
			res.num=(res.num+now.num)%MOD;
			res.sum=(res.sum+now.sum)%MOD;
			res.sum=(res.sum+i*inv[x]%MOD*now.num%MOD)%MOD;
		}
	}
	if (!limit && !lead)
	{
		dp[x][p]=res.num;
		f[x][p]=res.sum;
	}
	return res;
}

inline int solve(int x)
{
	memset(dp,-1,sizeof(dp));
	memset(f,0,sizeof(f));
	int len=0;
	while (x)
	{
		a[++len]=x%10;
		x/=10;
	}
	return dfs(len,-10,0,0,1,1).sum;
	//当前位，上一位数字，当前状态，已有多少数码，lead,limit 
}

signed main(){
	scanf("%lld%lld%lld",&n,&m,&k);
	inv[1]=1;
	for (register int i=2; i<=20; ++i) inv[i]=inv[i-1]*10%MOD; 
	printf("%lld\n",((solve(m)-solve(n-1))%MOD+MOD)%MOD);
return 0;	
}