HDU 5898 基础数位DP

来源：互联网发布：python genetic 编辑：程序博客网时间：2024/04/30 20:52

http://acm.hdu.edu.cn/showproblem.php?pid=5898

题意：统计连续的奇数必须是偶数个，连续的偶数必须是奇数个。。的个数

思路：dp[ i ] [ j ] [ k ] 前i位上一位是j 连续的奇数或者偶数有k。。

blog里面数位dp比较少，贴一个上来。。。

代码：

#include <stdio.h>#include <time.h>#include <stdlib.h>#include <iostream>#include <cstring>using namespace std;typedef long long ll;int a[20];ll dp[20][11][20];//不同题目状态不同ll dfs(int pos,int state,int ok,bool lead/*前导零*/,bool limit/*数位上界变量*/){    if(pos==-1&&(state+ok)%2==1) return 1;    if(pos==-1&&(state+ok)%2==0) return 0;    if(!limit && !lead && dp[pos][state][ok]!=-1) return dp[pos][state][ok];    int up=limit?a[pos]:9;    ll ans=0;    if(lead){        for(int i=0;i<=up;i++){            ans+=dfs(pos-1,i,1,lead && i==0,limit && i==a[pos]);        }    }    else if((state+ok)%2==1){        for(int i=0;i<=up;i++){            if((i+state)%2==0)                ans+=dfs(pos-1,i,ok+1,lead && i==0,limit && i==a[pos]);            else                ans+=dfs(pos-1,i,1,lead && i==0,limit && i==a[pos]);        }    }    else{        for(int i=0;i<=up;i++){            if((i+state)%2==0)                ans+=dfs(pos-1,i,ok+1,lead && i==0,limit && i==a[pos]);        }    }    if(!limit && !lead) dp[pos][state][ok]=ans;    return ans;}ll solve(ll x){    int pos=0;    a[0]=0;    while(x)    {        a[pos++]=x%10;//编号为[0,pos),        x/=10;    }    return dfs(pos-1/*从最高位开始枚举*/,10,1,true,true);}int main(){    ll le,ri;    int cas=1;    int t;    scanf("%d",&t);    memset(dp,-1,sizeof(dp));    while(t--){        scanf("%lld%lld",&le,&ri);        printf("Case #%d: %lld\n",cas++,solve(ri)-solve(le-1));    }}

0 0