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;
}