数位DP YbtOJ B数计数-CSDN博客

本文链接：https://blog.csdn.net/glorious_dream/article/details/123041167

考虑在dfs中传4个参数：pos，res，op，limit，分别表示位数，除以13的余数，是否出现13，当前位是否有取值限制

先来看op，有三类，0，1，2，分别表示没有13，有1，有13。

这时考虑用一个check函数

int check(int op,int x){
	if(op==0){
		if(x==1) return 1; //有1
		return 0; //什么都没有
	}
	else if(op==1){ //在有1的情况下
		if(x==3) return 2; //如果在它后面还有一个3
		if(x==1) return 1;
		return 0; 
	}
	else return 2; //op=2的时候已经满足条件了
}

然后就是数位DP的版子了

放上代码

#include<bits/stdc++.h>
using namespace std;
inline int read(){
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch == '-') f=-1 ; ch=getchar();}
	while(ch>='0'&&ch<='9'){x=(x<<1)+(x<<3)+(ch^48) ; ch=getchar();}
	return x*f;
}
int f[20][20][3];
int num[20];
int n;
int check(int op,int x){
	if(op==0){
		if(x==1) return 1;
		return 0;
	}
	else if(op==1){
		if(x==3) return 2;
		if(x==1) return 1;
		return 0;
	}
	else return 2;
}
int dfs(int pos,int res,int op,int limit){
	if(!limit && f[pos][res][op]!=-1) return f[pos][res][op]; //记忆化
	if(!pos) return res==0&&op==2; //所有位都搜完，看是否满足要求
	int ans=0;
	int s = limit ? num[pos] : 9; //下一位可以取到的数的范围
	for(register int i(0) ; i<=s ; i=-~i) ans+=dfs(pos-1,(res*10+i)%13,check(op,i),limit&&i==s); //往下搜
	if(!limit) f[pos][res][op] = ans; //没有limit的话记录
	return ans;
}
int calc(int x){
	int cnt=0;
	memset(f,-1,sizeof(f)); //别忘了一开始清成-1，因为有些题可能是0
	while(x) num[++cnt]=x%10,x/=10;
	return dfs(cnt,0,0,1);
}
int main(){
	while(scanf("%d",&n)!=EOF) printf("%d\n",calc(n)); 
	return 0;
}