hdu5787数位dp

题目：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;
}