[数位dp] uestc 250 windy数

题意:不含前导零且相邻两个数字之差至少为2的正整数被称为windy数。

思路: dp[i][j][k][f] i位 前一位是j  k代表是否符合条件了 f 代表是否含有前导0

代码:

#include"cstdlib"
#include"cstdio"
#include"cstring"
#include"cmath"
#include"queue"
#include"algorithm"
#include"iostream"
#define eps 1e-8
using namespace std;
int dp[12][12][3][2],num[12];
int dfs(int site,int sum,int ok,int f,int ff)
{
    if(site==0)
    {
        if(ff==0) return 0;
        else return ok?1:0;
    }
    if(!f&&dp[site][sum][ok][ff]!=-1) return dp[site][sum][ok][ff];
    int len=f?num[site]:9,ans=0;
    for(int i=0; i<=len; i++)
    {

        if(ok==0) ans+=dfs(site-1,i,0,f&&i==len,ff||i!=0);
        else
        {
            if(!ff) ans+=dfs(site-1,i,1,f&&i==len,ff||i!=0);
            else
            {
                if(abs(i-sum)<2) ans+=dfs(site-1,i,0,f&&i==len,ff||i!=0);
                else ans+=dfs(site-1,i,1,f&&i==len,ff||i!=0);
            }
        }
    }
    if(!f) dp[site][sum][ok][ff]=ans;
    return ans;
}
int solve(int x)
{
    int cnt=0;
    if(x<10) return x;
    while(x)
    {
        num[++cnt]=x%10;
        x/=10;
    }
    return dfs(cnt,99,1,1,0);  //注意最高位的前一位要保证成立
}
int main()
{
    int x,y;
    memset(dp,-1,sizeof(dp));

    while(scanf("%d%d",&x,&y)!=-1)
    {
        printf("%d\n",solve(y)-solve(x-1));
    }
}


你可能感兴趣的:(动态规划)