Sakura數

Sakura數

數位DP板子：(不解釋，懂得拿去用，改自經典數位DP入門題windy數)

#include<bits/stdc++.h>
using namespace std;
#define LL long long
LL dp[20][20];
LL a[20];
int INIT()
{
    memset(dp,0,sizeof(dp));
    memset(a,0,sizeof(a));
    LL i,j,k;
    for(i=0;i<10;i++)
        dp[1][i]=1;
    for(i=2;i<19;i++)    //預處理。
    {
        for(j=0;j<10;j++)   //第i位的數。
        {
            for(k=0;k<10;k++)//第i-1位的數。
            {
                if(fabs(j-k)>=2)
                    dp[i][j]+=dp[i-1][k];
            }
        }
    }
}
LL DP(LL n)
{
    LL i,j,k,top=0;
    while(n)
    {
        a[++top]=n%10;
        n=n/10;
    }
    LL sum=0;
    a[top+1]=0;
    for(i=1;i<top;i++)
        for(j=1;j<10;j++)
            sum+=dp[i][j];//先把小於最大位數的那些爲加上，因爲這裏不用考慮邊界條件 
    for(i=1;i<a[top];i++)
    {
        sum+=dp[top][i];
    }
    for(i=top-1;i;i--)
    {
        for(j=0;j<a[i];j++)
        {
            if(fabs(j-a[i+1])>=2)
                sum+=dp[i][j];
        }
        if(fabs(a[i+1]-a[i])<2)
            break;
    }
    return sum;
}
main()
{
    LL n,m,i,j,k;
    INIT();
    while(cin>>n>>m)
    {
        cout<<DP(m+1)-DP(n)<<endl;
    }
    return 0;
}