题目:http://acm.hdu.edu.cn/showproblem.php?pid=5787
分析:数位dp板子题,关键要去除前导0的影响。
ac code:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll dp[20][11][11][11][11];
int num[20];
int K;
inline bool differ(int n1,int n2,int n3,int n4,int i)
{
if(K==2) return i!=n1;
if(K==3) return i!=n1&&i!=n2;
if(K==4) return i!=n1&&i!=n2&&i!=n3;
if(K==5) return i!=n1&&i!=n2&&i!=n3&&i!=n4;
}
ll dfs(int pos,int n1,int n2,int n3,int n4,bool limit)
{
if(pos<0) return 1;
if(!limit&&~dp[pos][n1][n2][n3][n4]) return dp[pos][n1][n2][n3][n4];
int up_bound=limit?num[pos]:9;
ll ans=0;
for(int i=0;i<=up_bound;i++)
{
if(n1+n2+n3+n4==40&&i==0)///除去前导0的影响
ans+=dfs(pos-1,n1,n2,n3,n4,i==up_bound&&limit);
else if(differ(n1,n2,n3,n4,i))
ans+=dfs(pos-1,i,n1,n2,n3,i==up_bound&&limit);
}
if(!limit) dp[pos][n1][n2][n3][n4]=ans;
return ans;
}
ll solve(ll n)
{
int k=0;
while(n){
num[k++]=n%10;
n/=10;
}
return dfs(k-1,10,10,10,10,1);
}
int main()
{
ll l,r;
while(cin>>l>>r>>K){
memset(dp,-1,sizeof dp);
cout<<solve(r)-solve(l-1)<<endl;
}
return 0;
}