【codeforces 1036C】Classy Numbers(数位dp)

数位dp即可，记忆化一下,dfs传入三个变量:pos(数位)，num(当前还差几个数就能达到3个非0数),limit(是否到达上界)。

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e6+10;
const int INF = 0x3f3f3f3f;
typedef long long ll;
typedef int INT ;
#define int long long
int digit[20];
int dp[20][30];
int dfs(int pos,int num,int limit){
	if(pos<0&&num>=0) return 1;
	if(pos<0||num<0) return 0;
	if(!limit&&dp[pos][num]) return dp[pos][num];

	int ans=0;
	int up=limit?digit[pos]:9;

	for(int i=0;i<=up;i++){
		ans+=dfs(pos-1,num-(i?1:0),limit&&i==up);
	} 
	if(!limit) dp[pos][num]=ans;
	return ans;
}

int solve(int x){
	int cnt=0;
	while(x){
		digit[cnt++]=x%10;
		x/=10;
	}
	return dfs(cnt-1,3,1);
}

INT main(){
	int t,a,b;
	cin>>t;
	while(t--){
		memset(dp,0,sizeof(dp));
		cin>>a>>b;
		cout<<solve(b)-solve(a-1)<<endl;
	}
	return 0;
}