数位DP详解

最新推荐文章于 2022-04-15 11:31:05 发布

Wu_while

最新推荐文章于 2022-04-15 11:31:05 发布

阅读量809

点赞数 4

分类专栏：算法文章标签：动态规划

本文链接：https://blog.csdn.net/Wu_while/article/details/118975110

版权

算法专栏收录该内容

7 篇文章 0 订阅

订阅专栏

本文介绍了数位动态规划（DP）在解决区间内满足特定条件的整数计数问题上的方法。通过举例洛谷P2602题目，展示了如何在O(n)的时间复杂度内计算[1,a-1]和[1,b]之间满足条件的数的个数。在处理数据范围如1≤a≤b≤10^12的问题时，采用数位DP和前缀和思想，对数字进行位数构造。注意在处理前导0的情况时，需要特别考虑构造过程。

摘要由CSDN通过智能技术生成

~~好难啊~~

一、概念

数位DP，顾名思义，就是在数位上进行DP，对于一类“在区间 $[l, r]$ 内求满足条件的数的个数”的问题，显然，进行O(n)的枚举是不可能的，这时我们就要使用数位DP，在 $[l, r]$ 范围内进行构造满足条件的数。

二、实现

直接上例题。。。
洛谷 P2602

看一看数据范围：
$1≤a≤b≤10^{12}$
于是考虑数位DP。
我们要分别求出 $[1, b]$ 和 $[1, a - 1]$ 中满足条件的数，然后利用前缀和的思想来求出 $[a, b]$ 的答案。

$C o d e$

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
long long a,b;
long long pow[20];
long long f[20];
long long ansa[20],ansb[20];
long long num[20];
void dp(long long x,long long *ans)
{
	memset(num,0,sizeof(num));
	long long tot=0;
	while(x)
	{
		num[++tot]=x%10;
		x/=10;
	}
	for(long long i=tot;i>=1;i--)
	{
		for(long long j=0;j<=9;j++)
			ans[j]+=f[i-1]*num[i];
		for(long long j=0;j<num[i];j++)
			ans[j]+=pow[i-1];
		long long s=0;
		for(long long j=i-1;j>=1;j--)
			s=s*10+num[j];
		ans[num[i]]+=s+1;
		ans[0]-=pow[i-1];
	}
}
int main()
{
	scanf("%lld%lld",&a,&b);
	pow[0]=1;
	for(long long i=1;i<=13;i++)
	{
		f[i]=f[i-1]*10+pow[i-1];
		pow[i]=pow[i-1]*10;
	}
	dp(a-1,ansa);
	dp(b,ansb);
	for(long long i=0;i<=9;i++)
		printf("%lld ",ansb[i]-ansa[i]);
	return 0;
}

三、注意

在求 $[1, n]$ 范围内满足条件的数的个数时，我们一般在 $n$ 的位数范围内进行构造，比方说，我们要求 $[1, 1234]$ 内的答案，我们就构造所有 $4$ 位的答案，从高位到低位进行构造，允许有前导 $0$ 。那又如何构造小于等于 $n$ 的答案呢？我们在从高位到低位枚举的过程中，记录下当前位小于或等于 $n$ 的对应位数。若小于，则比当前位更低的所有位数都不在受 $n$ 的限制；若等于，则低位继续受限制。
有些题目会限制前导 $0$ ，这时就要用变量记录下前导 $0$ 。