hdu 5898 (数位dp)

来源:互联网 发布:js 跳转url 隐藏参数 编辑:程序博客网 时间:2024/05/17 03:15


题意:统计数位上的连续的奇数个数都是偶数,偶数个数为奇数的数  的个数。

思路:一看就知道是数位dp,当初一会就写出来了,然后调bug调了两个钟头,还是太菜。

        用两个数记录奇数和偶数的个数,当一个与之前不同的数加入的时候,另一种数要清零。

        要记录是否是前导零,因为前导零关乎着加入的加入的零算不算数。

     

#include<iostream>#include<cstdio>#include<cstring>#include<cmath>#include<vector>#include<algorithm>#include<map>#include<set>#include<queue>using namespace std;const int maxn = 20 + 5;#define INF 0x3f3f3f3f#define clr(x,y) memset(x,y,sizeof x )typedef long long ll;#define eps 10e-8const int Mod = 1e9 + 2;typedef pair<int, int> P;int bits[maxn];ll dp[maxn][maxn][maxn];int a[maxn];ll dfs(int pos,bool lead,int cnt1,int cnt2,bool flag)//cnt1 连续奇数的个数,cnt2 连续偶数的个数{    if(pos < 0)return (cnt1 % 2 == 0 && (cnt2 % 2 || cnt2 == 0));    if(!flag && !lead && dp[pos][cnt1][cnt2] != -1)return dp[pos][cnt1][cnt2];    int up = flag ? bits[pos] : 9;    ll ret = 0;    for(int i = 0; i <= up; i ++)    {        if(i & 1)//奇数        {            if( (cnt2 % 2) || (cnt2 == 0) )            {                a[pos] = i;                ret += dfs(pos - 1,lead && i == 0,cnt1 + ( (lead && i == 0) ? 0 : 1 ),0,flag && i == up);            }        }        else        {            if( cnt1 % 2 == 0 )            {                a[pos] = i;                ret += dfs(pos - 1,lead && i == 0,0,cnt2 + ( (lead && i == 0) ? 0 : 1 ),flag && i == up);            }        }    }    if(!lead && !flag)dp[pos][cnt1][cnt2] = ret;    return ret;}ll calc(ll n){    int len = 0;while(n){bits[len ++] = n % 10;n /= 10;}    return dfs(len - 1,true,0,0,true);}int main(){    ll n,m;clr(dp,-1);    int Tcase;scanf("%d",&Tcase);    for(int ii = 1; ii <= Tcase; ii ++)    {        scanf("%I64d%I64d",&n,&m);        printf("Case #%d: %I64d\n",ii,calc(m) - calc(n - 1));    }    return 0;}