hdu5898(数位dp)

链接：点击打开链接

题意：求L到R区间内，连续奇数个数是偶数，连续偶数个数是奇数的数的个数

代码：

#include <map>
#include <set>
#include <queue>
#include <string>
#include <math.h>
#include <vector>
#include <stdio.h>
#include <stdlib.h>
#include <iostream>
#include <string.h>
#include <algorithm>
using namespace std;
int a[20];
long long dp[20][20][20][20];
long long dfs(int pos,int pre,int sta1,int sta2,int lim,int fig){
    int i,u;
    long long ans=0;
    if(pos==0){
        if((sta1%2==0&&sta2==0)||(sta2%2!=0&&sta1==0))
        return 1;
        return 0;
    }
    if(dp[pos][pre][sta1][sta2]!=-1&&lim==0)
    return dp[pos][pre][sta1][sta2];
    u=lim?a[pos]:9;
    for(i=0;i<=u;i++){
        if(i%2!=0){
            if(fig==1&&i==0)
            ans+=dfs(pos-1,i,0,0,lim&(i==u),1);
            else if(sta2%2==1||sta2==0)
            ans+=dfs(pos-1,i,sta1+1,0,lim&(i==u),0);
        }
        else{                                   //要注意一下不要把前导0加入计数中
            if(fig==1&&i==0)
            ans+=dfs(pos-1,i,0,0,lim&(i==u),1);
            else if(sta1%2==0||sta1==0)
            ans+=dfs(pos-1,i,0,sta2+1,lim&(i==u),0);
        }
    }
    if(lim==0)
    dp[pos][pre][sta1][sta2]=ans;
    return ans;                                 //记录一下奇数和偶数连续出现的个数
}
long long cal(long long x){
    int pos=0;
    while(x){
        a[++pos]=x%10;
        x/=10;
    }
    return dfs(pos,0,0,0,1,1);
}
int main(){
    int T,cas;
    long long l,r;
    scanf("%d",&T);
    memset(dp,-1,sizeof(dp));
    for(cas=1;cas<=T;cas++){
        scanf("%I64d%I64d",&l,&r);
        printf("Case #%d: %I64d\n",cas,cal(r)-cal(l-1));
    }
    return 0;
}