Submission #579895

#TimeUsernameProblemLanguageResultExecution timeMemory
579895RealSnakePalindrome-Free Numbers (BOI13_numbers)C++14
100 / 100
1 ms372 KiB
#include "bits/stdc++.h"
using namespace std;

#define ll long long
#define mod 1000000007

#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace __gnu_pbds;
typedef tree<ll, null_type, less<ll>, rb_tree_tag, tree_order_statistics_node_update> ordered_set;

ll n;
string s;
ll dp[20][2][11][11][2];

ll solve(ll ind, bool b, ll lst1, ll lst2, bool zero) {
    if(ind + 1 == n)
        return 1;
    if(dp[ind + 1][b][lst1][lst2][zero] != -1)
        return dp[ind + 1][b][lst1][lst2][zero];
    ll ans = 0;
    if(b == 1) {
        ll d = (s[ind + 1] - '0');
        for(ll i = 0; i <= d; i++) {
            if(!zero && (i == lst1 || i == lst2))
                continue;
            if(zero && (i == 0))
                ans += solve(ind + 1, (i == d), 10, 10, 1);
            else
                ans += solve(ind + 1, (i == d), i, lst1, (zero && (i == 0)));
        }
    }
    else {
        for(ll i = 0; i <= 9; i++) {
            if(!zero && (i == lst1 || i == lst2))
                continue;
            if(zero && (i == 0))
                ans += solve(ind + 1, 0, 10, 10, 1);
            else
                ans += solve(ind + 1, 0, i, lst1, (zero && (i == 0)));
        }
    }
    return dp[ind + 1][b][lst1][lst2][zero] = ans;
}

signed main() {

    ios::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    string b;
    ll a;
    cin >> a >> b;
    s = b;
    n = s.size();
    memset(dp, -1, sizeof dp);
    ll ans = solve(-1, 1, 10, 10, 1);
    a--;
    if(a >= 0) {
        memset(dp, -1, sizeof dp);
        s = to_string(a);
        n = s.size();
        ans -= solve(-1, 1, 10, 10, 1);
    }
    cout << ans;

    return 0;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...