世界真的很大
前几天考试考了个数位DP
当时真的是一脸蒙蔽
于是乎花了时间好好研究了一下这道题
可能也并不是这么难
有大佬指路
先看下题
description
给定一个数n,求1到n范围内有多少个数既含有13又是13的倍数
input
多组数据,到EOF
每组数据包含一个整数n
output
每组数据对应一个整数,即满足条件的数的个数
首先看一下,确定题目类型数位dp无误
但是经过我的研究
这个数位dp,其实dp的性质不算很明显,硬要说的话,还是叫做记忆化搜索吧。主要思路就是直接按照数位dfs就行
考虑dfs维护的状态
从高到低的数位,这个数的前几位mod13的值,枚举的数位的前一位的数字(因为要包含13,前一位是1,这一位是3就行了,所以只需要记录前一位是不是1)
尽管是从高到低枚举数位,但因为是从0开始枚举每一位的数字,所以允许前导0的存在,如果有前导0的话,就相当于少了几位
题目要求是小于n
所以枚举的数必然从最高位往低数某一位就小于n的同一位
也就是说在枚举的数的那一位到来前,每一位枚举的数都有上限,就是原数n的同一位
如果枚举的数从某一位开始就比原数小的话,之后枚举的数就没有限制了,从0–9
每次注意保留has的状态
0表示当前枚举的前一位不是1
1表示是1
2表示之前枚举的数位里已经有13了
每次枚举当前数位的时候特判has的状态,继续dfs
mod的值的转移相当于做除法竖式时的感觉,把前几位的mod乘以10加上当前位枚举的数再mod13
最后在枚举完所有数位的时候,如果mod为0说明按位枚举的这个数是成立的,返回1
用一个三维数组f[i][j][k],位数,mod,状态来记忆化搜索
完整代码:
#include<stdio.h>
#include<cstring>
#include<algorithm>
using namespace std;
int f[15][15][10],bit[15];
int n;
int dfs(int pos,int mod,int has,int lim)
{
if(pos<=0) return has==2&&mod==0;
if(!lim&&f[pos][mod][has]!=-1) return f[pos][mod][has];
int num=lim ? bit[pos] : 9 ;
int ans=0,mod_x,has_x;
for(int i=0;i<=num;i++)
{
has_x=has;
mod_x=(mod*10+i)%13;
if(has==0 && i==1) has_x=1;
if(has==1 && i!=1) has_x=0;
if(has==1 && i==3) has_x=2;
ans+=dfs(pos-1,mod_x,has_x,lim&&i==num);
}
if(!lim) f[pos][mod][has]=ans;
return ans;
}
int main()
{
while(scanf("%d",&n)!=EOF)
{
memset(f,-1,sizeof(f));
memset(bit,0,sizeof(bit));
int len=0;
while(n)
{
bit[++len]=n%10;
n/=10;
}
printf("%d\n",dfs(len,0,0,1));
}
}
嗯,就是这样