Submission #474778

#	Time	Username	Problem	Language	Result	Execution time	Memory
474778		Duy_e	Palindrome-Free Numbers (BOI13_numbers)	C++14	100 / 100	2 ms	332 KiB

numbers

#include <bits/stdc++.h>
#define ll long long
#define ull unsigned long long
#define pii pair<long long, long long>
#define st first
#define nd second
#define file "test"
using namespace std;
const long long oo = 1e18;
const long long N = 2e5 + 5;

ll dp[23][11][11][2];
vector <int> num;

ll cal(int pos, int pre, int cur, bool f){

    if (pos == num.size()) return 1;

    ll &ans = dp[pos][pre][cur][f];
    if (ans != - 1) return ans;
    ans = 0;

    ll lmt = num[pos];
    if (f) lmt = 9;

    for (int i = 0; i <= lmt; i ++)
    if (i != pre && i != cur){
        bool new_f = f;
        if (f == 0 && i < lmt) new_f = 1;
        ans += cal(pos + 1, cur, (cur == 10 && i == 0? 10 : i), new_f);
    }
    return ans;
}

ll solve(ll n){
    memset(dp, -1, sizeof dp);
    if (n < 0) return 0;
    num.clear();

    while (n > 0) num.push_back(n%10), n/=10;
    reverse(num.begin(), num.end());
    return cal(0, 10, 10, 0);
}

int main()
{
    ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
    
    ll A, B;
    cin >> A >> B;
    cout << solve(B) - solve(A - 1);

    return 0;
}    
/**  /\_/\
 *  (= ._.)
 *  / >0  \>1
**/

Compilation message (stderr)

numbers.cpp: In function 'long long int cal(int, int, int, bool)':
numbers.cpp:17:13: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   17 |     if (pos == num.size()) return 1;
      |         ~~~~^~~~~~~~~~~~~

• Subtask 1: 25 / 25
• Subtask 2: 75 / 75