hdu 3709+hdu 3555(数位dp)

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=3709

数位dp...完全看大牛模版理解的。。。。

#include<iostream>
using namespace std;
__int64 dp[19][19][2000];
int digit[19];

//pos表示当前的位置,o表示支点，pre表示从最高位到pos的力矩之和，doing表示是否有上限，若无，则为9;
__int64 dfs(int pos,int o,int pre,bool doing){
    if(pos==-1){
        return pre==0;  //已经全部组合
    }
    if(pre<0)return 0;//如果前面组合的力矩之和已经为负，则后面的必然小于0；
    //没有上限，且前面的状态已经搜过
    if(!doing&&dp[pos][o][pre]!=-1){
        return dp[pos][o][pre];
    }
    __int64 ans=0;
    int end=doing?digit[pos]:9;
    for(int i=0;i<=end;i++){
        int npre=pre;   //枚举下一个状态
        npre+=(pos-o)*i;   
        ans+=dfs(pos-1,o,npre,doing&&i==end);
    }
    if(!doing){
        dp[pos][o][pre]=ans;
    }
    return ans;
}

__int64 solve(__int64 n){
    int pos=0;
    while(n){
        digit[pos++]=n%10;
        n/=10;
    }
    __int64 ans=0;
    for(int o=0;o<pos;o++){
        ans+=dfs(pos-1,o,0,1);
    }
    return ans-(pos-1);//0,00,000....这种情况
}

int main(){
    int _case;
    scanf("%d",&_case);
    memset(dp,-1,sizeof(dp));
    while(_case--){
        __int64 n,m;
        scanf("%I64d%I64d",&n,&m);
        printf("%I64d\n",solve(m)-solve(n-1));
    }
    return 0;
}

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=3555

#include<iostream>
using namespace std;
__int64 dp[20][3];
int digit[20];

//have==1表示前一位为4,have==0表示没有出现49，have==2表示前面有49出现
__int64 dfs(int pos,int have,bool doing){
    if(pos==-1){
        return have==2;//找到一个解
    }
    if(!doing&&dp[pos][have]!=-1){
        return dp[pos][have];
    }
    __int64 ans=0;
    int end=doing?digit[pos]:9;
    for(int i=0;i<=end;i++){
        int nhave=have;
        if(have==1&&i!=4){
            nhave=0;
        }
        if(have==0&&i==4){
            nhave=1;
        }
        if(have==1&&i==9){
            nhave=2;
        }
        ans+=dfs(pos-1,nhave,doing&&i==end);
    }
    if(!doing)
        dp[pos][have]=ans;
    return ans;
}

__int64 solve(__int64 n){
    int pos=0;
    while(n){
        digit[pos++]=n%10;
        n/=10;
    }
    return dfs(pos-1,0,1);
}

int main(){
    int _case;
    scanf("%d",&_case);
    memset(dp,-1,sizeof(dp));
    while(_case--){
        __int64 n;
        scanf("%I64d",&n);
        printf("%I64d\n",solve(n));
    }
    return 0;
}

昨天做的fzu的月赛。。。当时没过。。。

http://acm.fzu.edu.cn/contest/problem.php?cid=126&sortid=1

这个代码我没提交过。。。

#include<iostream>
#include<cstring>
using namespace std;
__int64 dp[20][20];
int digit[20];

__int64 DFS(int pos,int have,bool doing){
    if(pos==-1){
        return have;
    }
    if(!doing&&dp[pos][have]!=-1){
        return dp[pos][have];
    }
    __int64 ans=0;
    int end=doing?digit[pos]:9;
    for(int i=0;i<=end;i++){
        int nhave=have;
        if(i==1)nhave=have+1;
        ans+=DFS(pos-1,nhave,doing&&i==end);
    }
    if(!doing){
        dp[pos][have]=ans;
    }
    return ans;
}


__int64 solve(__int64 n){
    int pos=0;
    while(n){
        digit[pos++]=n%10;
        n/=10;
    }
    return DFS(pos-1,0,1);
}

int main(){ 
    __int64 n,m;
    memset(dp,-1,sizeof(dp));
    while(~scanf("%I64d%I64d",&n,&m)){
        printf("%I64d\n",solve(m)-solve(n-1));
    }
    return 0;
}

        

PS:记忆化搜索真的很强大啊。。。