數位dp - 板子題

無愧於人發表於2020-11-12

題目傳送
存一個狀態就可以了,用來判斷前一位是不是6的情況
具體看程式碼註釋
AC程式碼

#include <bits/stdc++.h>
inline long long read(){char c = getchar();long long x = 0,s = 1;
while(c < '0' || c > '9') {if(c == '-') s = -1;c = getchar();}
while(c >= '0' && c <= '9') {x = x*10 + c -'0';c = getchar();}
return x*s;}
using namespace std;
#define NewNode (TreeNode *)malloc(sizeof(TreeNode))
#define Mem(a,b) memset(a,b,sizeof(a))
#define lowbit(x) (x)&(-x)
const int N = 2e5 + 10;
const long long INFINF = 0x7f7f7f7f7f7f7f;
const int INF = 0x3f3f3f3f;
const double EPS = 1e-5;
const int mod = 1e9+7;
const double II = acos(-1);
const double PP = (II*1.0)/(180.00);
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> pii;
typedef pair<ll,ll> piil;
ll dp[20][2],a[20];
ll dfs(ll pos,ll pre,ll sta,bool limit)
{
    if(pos == -1) return 1;//當小於0時,返回1的原因是上一個位置的這個數算一次
    if(!limit && dp[pos][sta] != -1) return dp[pos][sta];//當沒有限制條件的時候並且這個狀態下的這個位置的數目已經在前面記憶化過,那麼直接加
    ll up = limit ? a[pos] : 9;//如果沒有限制,就是最高位9,否則在有限制條件下,那麼這個位置的最高數字就是a[pos]這一位的最大數
    ll temp = 0;//這個位置這個狀態下的計數
    for(int i = 0;i <= up;i++)
    {
        if((pre == 6 && i == 2) || i == 4) continue;
        temp += dfs(pos-1,i,i == 6,limit && i == a[pos]);//狀態轉移
    }
    if(!limit) dp[pos][sta] = temp;//記憶化
    return temp;//否則有限制的話就不用記憶化,因為只判斷一次有限制的時候
}
ll solve(ll num)
{
    ll pos = 0;
    while(num)
    {
        a[pos++] = num%10;
        num /= 10;
    }
    ll sum = dfs(pos-1,-1,0,true);
//    for(int i = pos-1;i >= 0;i--)
//        cout << dp[i][0] << " " << dp[i][1] << endl;
    return sum;
}
signed main()
{
    std::ios::sync_with_stdio(false);
    cin.tie(0),cout.tie(0);
    //    freopen("input.txt","r",stdin);
    //    freopen("output.txt","w",stdout);
    int l,r;
    while(cin >> l >> r)
    {
        if(!l && !r) break;
        memset(dp,-1,sizeof(dp));
        cout << solve(r)-solve(l-1) << endl;
    }
}