UESTC 1307 windy数(数位DP)

题目链接

这其实入门题，不过，我写了好一会，还是一直wa，图样。。。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <map>
#include <ctime>
#include<sstream>
#include <cmath>
#include <algorithm>
using namespace std;
int dp[11][11],sum[11];
int judge(int x)
{
    int num[11],i,j,n = 0,pre;
    int ans = 0;
    if(x == 0)
    return 1;
    while(x)
    {
        num[n++] = x%10;
        x /= 10;
    }
    if(n == 1)
    {
        return num[0] + 1;
    }
    for(i = 1;i < n;i ++)
    {
        for(j = 1;j < 10;j ++)
        ans += dp[i][j];
    }
    ans ++;
    for(i = 1;i < num[n-1];i ++)
    {
        ans += dp[n][i];
    }
    pre = num[n-1];
    for(i = n-2; i >= 0; i --)
    {
        if(i == 0)
        {
            for(j = 0; j <= num[i]; j ++)
            {
                if(max(j,pre)-min(j,pre) >= 2)
                ans ++;
            }
            break;
        }
        for(j = 0; j < num[i]; j ++)
        {
            if(max(j,pre)-min(j,pre) >= 2)
            {
                ans += dp[i+1][j];
            }
        }
        if(max(num[i],pre)-min(num[i],pre) < 2) break;
        pre = num[i];
    }
    return ans;
}

int main()
{
    int i,j,k,x,y;
    for(i = 0; i < 10; i ++)
        dp[1][i] = 1;
    for(i = 2; i <= 10; i ++)
    {
        for(j = 0; j < 10; j ++)
        {
            for(k = 0; k < 10; k ++)
            {
                if(max(j,k)-min(j,k) >= 2)
                {
                    dp[i][j] += dp[i-1][k];
                }
            }
        }
    }
    while(scanf("%d%d",&x,&y)!=EOF)
    {
        printf("%d
",judge(y)-judge(x-1));
    }
    return 0;
}