动态规划——数位dp

数位dp

文章目录

  • 数位dp
    • 概述
      • 题目特征
      • 基本原理
      • 计数技巧
    • 模板
    • 例题
      • 度的数量
        • 思路
        • 代码
      • 数字游戏
        • 思路
        • 代码
      • 不要62
        • 思路
        • 代码

概述

数位是指把一个数字按照个、十、百、千等等一位一位地拆开,关注它每一位上的数字。如果拆的是十进制数,那么每一位数字都是 0~9,其他进制可类比十进制。

题目特征

数位 DP:用来解决一类特定问题,这种问题比较好辨认,一般具有这几个特征:

  1. 要求统计满足一定条件的数的数量(即,最终目的为计数);

  2. 这些条件经过转化后可以使用「数位」的思想去理解和判断;

  3. 输入会提供一个数字区间(有时也只提供上界)来作为统计的限制;

  4. 上界很大(比如 ),暴力枚举验证会超时。

基本原理

考虑人类计数的方式,最朴素的计数就是从小到大开始依次加一。但我们发现对于位数比较多的数,这样的过程中有许多重复的部分。例如,从 7000 数到 7999、从 8000 数到 8999、和从 9000 数到 9999 的过程非常相似,它们都是后三位从 000 变到 999,不一样的地方只有千位这一位,所以我们可以把这些过程归并起来,将这些过程中产生的计数答案也都存在一个通用的数组里。此数组根据题目具体要求设置状态,用递推或 DP 的方式进行状态转移。

计数技巧

数位 DP 中通常会利用常规计数问题技巧,比如把一个区间内的答案拆成两部分相减(即 a n s [ l , r ] = a n s [ 0 , r ] − a n s [ 0 , l − 1 ] \mathit{ans}_{[l, r]} = \mathit{ans}_{[0, r]}-\mathit{ans}_{[0, l - 1]} ans[l,r]=ans[0,r]ans[0,l1]

模板

动态规划——数位dp_第1张图片

# 假设n为b进制数
def dp(n) :
	# 特判数为0,防止下面把每位提出时的bug
	if not n : return ..
	nums = []
	while n :
		nums.append(n % b)
		n //= b
	res, last = 0, 0 #分别存储上结果和前面位的对当前位的有用信息
	for i in range(len(nums) - 1, -1, -1) : #对每一位进行枚举
		for j in 除上界以外可能的数 :
			if 进行可行性判断 :
				res += 预处理的数
		if 取上界不合法: break
		#当第i为取上界时,判断可行性再进行下一轮
		if not i and 判断是否可行 : res += 1
	return res

例题

度的数量

求给定区间 [X,Y] 中满足下列条件的整数个数:这个数恰好等于 K 个互不相等的 B 的整数次幂之和。

例如,设 X=15,Y=20,K=2,B=2,则有且仅有下列三个数满足题意:

17=24+20
18=24+21
20=24+22
输入格式
第一行包含两个整数 X 和 Y,接下来两行包含整数 K 和 B。

输出格式
只包含一个整数,表示满足条件的数的个数。

数据范围
1≤X≤Y≤231−1,
1≤K≤20,
2≤B≤10
输入样例:
15 20
2
2
输出样例:
3

思路

分类讨论,当枚举的位数处于第i位时 x i x_i xi的情况,当N的第i位为x:

  1. 当x = 0时:则 x i x_i xi只能取0,即上界的情况,则继续向下枚举即可
  2. 当x = 1时:则 x i x_i xi取上界以外的情况只能是取0,此时低位的数可以随意填写,但必须满足题目中1的总个数等于k的前提下。随后取 x i = 1 x_i = 1 xi=1,继续向低位枚举。
  3. 当x > 1时:则 x i x_i xi取上界以外的情况只可以是0、1,此时低位的数可以随意填写,但必须满足题目中1的总个数等于k的前提下。但 x i x_i xi永远碰不到上界,则无需向低位枚举。

注意:当枚举到第i位时,已经使用了last个1,如果是未取上界的情况,那么剩下的i位低位中剩余k-last个1可用随便放。是个组合数,可以预处理出来。

代码

N = 35
f = [[0] * N for _ in range(N)]

def init() : #预处理出组合数
	for i in range(N) :
		for j in range(N) :
			if j == 0 :
				f[i][j] = 1
			else :
				f[i][j] = f[i - 1][j - 1] + f[i - 1][j]

def dp(n) :
	# 特判0
	if not n : return 0
	nums = []
	#处理出n在b进制下的每一位的数
	while n : 
		nums.append(n % b)
		n //= b
	res, last = 0, 0 #分别记录结果和高位对1的使用情况
	for i in range(len(nums) - 1, -1, -1) : #从高位往低位枚举
		x = nums[i] #取出n中第i位的数
		if x : #当0不是上界
			res += f[i][k - last] #第i位为0的情况
			#当上界大于1的情况
			if x > 1 :
				if k - last - 1 >= 0 : 
					res += f[i][k - last - 1]
				break
			# 取上界为1时
			else :
				last += 1
				if last > k : break
		if not i and last == k : res += 1
	return res
	
init()			
l, r = map(int, input().split())
k = int(input())
b = int(input())
print(dp(r) - dp(l - 1))

数字游戏

科协里最近很流行数字游戏。

某人命名了一种不降数,这种数字必须满足从左到右各位数字呈非下降关系,如 123,446。

现在大家决定玩一个游戏,指定一个整数闭区间 [a,b],问这个区间内有多少个不降数。

输入格式
输入包含多组测试数据。

每组数据占一行,包含两个整数 a 和 b。

输出格式
每行给出一组测试数据的答案,即 [a,b] 之间有多少不降数。

数据范围
1≤a≤b≤231−1
输入样例:
1 9
1 19
输出样例:
9
18

思路

分类讨论,当枚举的位数处于第i位时 x i x_i xi的情况,当N的第i位为x:

  1. x i = x x_i = x xi=x时,即取上界时,如果合法,则继续向更低位开始枚举。
  2. x i < x x_i < x xi<x时,则更低位的数应该是所有以 [ l a s t , x ) [last, x) [last,x)为最高位的i + 1位数的非下降的数

last为上一位的上界即 x i + 1 x_{i + 1} xi+1

预处理:
状态表示:
集合:f[i, j]表示最高位为j的i位的非下降的数的集合
属性:num
状态计算: f [ i , j ] = ∑ j 9 f [ i − 1 , k ] f [i, j] = \sum_j^9 f[i - 1, k] f[i,j]=j9f[i1,k]

代码

N = 15

f = [[0] * N for _ in range(N)]

# 预处理出最高位为j的i位的非下降的数的数量
def init() :
	for i in range(10) : f[1][i] = 1
	for i in range(2, N) :
		for j in range(10) :
			for k in range(j, 10) :
				f[i][j] += f[i - 1][k]

def dp(n) :
	# 特判0,有一个数满足条件
	if not n : return 1
	nums = []
	#处理出n的每一位,存于nums中
	while n :
		nums.append(n % 10)
		n //= 10
	res, last = 0, 0 #last存储上一层的数
	# 从高到低枚举每一位
	for i in range(len(nums) - 1, -1, -1) :
		x = nums[i]
		if last > x : break #如果当前位能取到的最大的数小于上一层上界,则退出
		#x_i不取上界
		for j in range(last, x) :
			res += f[i + 1][j]
		#取上界
		last = x
		if not i : res += 1 #特判最后一个数
	return res
			
init()
while True :
	try :
		l, r = map(int, input().split())
	except : break
	print(dp(r) - dp(l - 1))

不要62

杭州人称那些傻乎乎粘嗒嗒的人为 62(音:laoer)。

杭州交通管理局经常会扩充一些的士车牌照,新近出来一个好消息,以后上牌照,不再含有不吉利的数字了,这样一来,就可以消除个别的士司机和乘客的心理障碍,更安全地服务大众。

不吉利的数字为所有含有 4 或 62 的号码。例如:62315,73418,88914 都属于不吉利号码。但是,61152 虽然含有 6 和 2,但不是 连号,所以不属于不吉利数字之列。

你的任务是,对于每次给出的一个牌照号区间 [n,m],推断出交管局今后又要实际上给多少辆新的士车上牌照了。

输入格式
输入包含多组测试数据,每组数据占一行。

每组数据包含一个整数对 n 和 m。

当输入一行为“0 0”时,表示输入结束。

输出格式
对于每个整数对,输出一个不含有不吉利数字的统计个数,该数值占一行位置。

数据范围
1≤n≤m≤109
输入样例:
1 100
0 0
输出样例:
80

思路

分类讨论,当枚举的位数处于第i位时 x i x_i xi的情况,当N的第i位为x:

  1. x i = x x_i = x xi=x时,即取上界时,如果合法(不等于4且last与 x i x_i xi不等于62),则继续向更低位开始枚举。
  2. x i < x x_i < x xi<x时,则更低位的数应该是所有以 [ 0 , x ) [0, x) [0,x)为最高位的i + 1位数合法的数

last为上一位的上界即 x i + 1 x_{i + 1} xi+1

预处理:
状态表示:
集合:f[i, j]表示最高位为j的i位的吉利的号码的集合
属性:num
状态计算: f [ i , j ] = ∑ 0 9 f [ i − 1 , k ] f [i, j] = \sum_0^9 f[i - 1, k] f[i,j]=09f[i1,k]

代码

N = 10

f = [[0] * N for _ in range(N)]
# 预处理 f[i, j]表示最高位为j的i位的吉利的号码的集合的数目
def init() :
	for i in range(10) :
		if i == 4 : continue
		f[1][i] = 1
	for i in range(2, N) :
		for j in range(10) :
			if j == 4 : continue
			for k in range(10) :
				if k== 4 or (j == 6 and k == 2) : continue
				f[i][j] += f[i - 1][k]

def dp(n) :
	if not n : return 1
	nums = []
	while n :
		nums.append(n % 10)
		n //= 10
	res, last = 0, 0
	# 从高到低枚举每一位
	for i in range(len(nums) - 1, -1, -1) :
		x = nums[i]
		# x_i不取上界
		for j in range(x) :
			# 判断是否合法
			if j == 4 or (last == 6 and j == 2) :
				continue
			res += f[i + 1][j]
		# 当取上界x不合法时,则退出
		if x == 4 or (last == 6 and x == 2) : break
		last = x
		# 特判数为n时
		if not i : res += 1
	return res
init()
while True :
	l, r = map(int, input().split())
	if l == 0 and r == 0 : break
	print(dp(r) - dp(l - 1))

你可能感兴趣的:(#,动态规划,动态规划,算法)