首页 > 代码库 > [数位dp] spoj 10738 Ra-One Numbers

[数位dp] spoj 10738 Ra-One Numbers

题意:给定x、y,为[x,y]之间有多少个数的偶数位和减去奇数位和等于一。

个位是第一位。

例子: 10=1-0=1 所以10是这样的数

思路:数位dp[i][sum][ok] i位和为sum 是否含有前导0.

然后就是因为有负数 所以根据范围把0设置为100 然后最后和等于101则为所求的数。

代码:

#include"cstdlib"
#include"cstdio"
#include"cstring"
#include"cmath"
#include"stack"
#include"algorithm"
#include"iostream"
using namespace std;
int dp[12][200][2],num[12];
int fuck[2]={1,-1};
int dfs(int site,int sum,int ok,int f)
{
    if(site==0)
    {
        if(ok==0) return 0;
        return sum==101?1:0;  //小处理 和为101
    }
    if(!f&&dp[site][sum][ok]!=-1) return dp[site][sum][ok];
    int len=f?num[site]:9;
    int ans=0;
    for(int i=0;i<=len;i++)
    {
        if(ok==0)
        {
            if(i==0) ans+=dfs(site-1,sum,ok||i!=0,f&&i==len);
            else ans+=dfs(site-1,sum+i*fuck[site%2],ok||i!=0,f&&i==len);
        }
        else
        {
            ans+=dfs(site-1,sum+i*fuck[site%2],ok||i!=0,f&&i==len);
        }
    }
    if(!f) dp[site][sum][ok]=ans;
    return ans;
}
int solve(int x)
{
    if(x<0) return 0;
    int cnt=0;
    while(x)
    {
        num[++cnt]=x%10;
        x/=10;
    }
    return dfs(cnt,100,0,1);  //进入的时候 sum=100
}
int main()
{
    int t;
    scanf("%d",&t);
    memset(dp,-1,sizeof(dp));
    while(t--)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        printf("%d\n",solve(y)-solve(x-1));
    }
    return 0;
}


[数位dp] spoj 10738 Ra-One Numbers