数位dp(模板)

一篇不错的博客

一般数位dp问题是这样的:

求出在给定区间 [A,B] 内,符合条件 f(i) 的数 i 的个数。条件 f(i) 一般与数的大小无关,而与数的组成有关

例如:给出一个数n,1~n有多少数包含49,测试数据1<=T<=10000,1<=n<=2^63-1

#include<bits/stdc++.h>
using namespace std;
const int Max = 99999;
const int Min = 0;
const int inf = 1e6;
const int mod = 1e9+7;
#define M 1000
#define N 1000
#define ll long long
#define swap(x,y) x^=y^=x^=y
ll dp[20][6];
int digit[20];
ll dfs(int pos,int pre,int sta,bool limit){  //以53421为例 
	if(pos==-1) return 1;                   //代表这种情况搜索结束,+1 
	if(!limit&&dp[pos][sta]!=-1) return dp[pos][sta];           //如果没有达到上限比如搜索0--50000 的时候,第一位是0,1,2,3,4的时候没有限制,5的时候有 
	int up = limit?digit[pos]:9;                                  //有限制的话选取下一位的值,如万位是0,1,2,3,4,千位可以是1-9,但是万位是5,千位不能超过3 
	ll sum(0); 
	for(int i=0;i<=up;i++){                                       //每一位进行枚举 
		if(pre==4&&i==9) continue;                                 //符合条件的不搜了 
		sum += dfs(pos-1,i,i==4,limit&&i==digit[pos]);             //不符合条件的加上,pre记录这一位的值,sta记录是否有可能成为49,最后一个表示是否有限制 
	}
	if(!limit) dp[pos][sta] = sum;                        //没有限制将dp的数值存起来,以便调用 
	return sum;                                        //返回值 
}
ll solve(ll a){
	int cnt = 0;                       //分解这个数
	while(a>0){              
		digit[cnt++] = a%10;
		a/=10;
	}
	ll ans = dfs(cnt-1,0,0,true);          //对这个数进行dfs
	return ans;
}
int main(){
	#ifdef LOCAL
		freopen("test.txt","r",stdin);
	#endif
	int T;
	cin >> T;
	memset(dp,-1,sizeof(dp));
	while(T--){
		ll a;
		scanf("%I64d",&a);
		ll ans = solve(a); 
		cout << a+1-ans << endl;
	}
    return 0;
}

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值