結果

問題 No.2455 Numbers Dictionary
ユーザー asaringoasaringo
提出日時 2023-09-01 23:55:04
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 72 ms / 2,000 ms
コード長 5,630 bytes
コンパイル時間 2,477 ms
コンパイル使用メモリ 208,392 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-06-11 06:13:50
合計ジャッジ時間 4,294 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 51 ms
5,376 KB
testcase_03 AC 44 ms
5,376 KB
testcase_04 AC 43 ms
5,376 KB
testcase_05 AC 50 ms
5,376 KB
testcase_06 AC 72 ms
5,376 KB
testcase_07 AC 65 ms
5,376 KB
testcase_08 AC 34 ms
5,376 KB
testcase_09 AC 67 ms
5,376 KB
testcase_10 AC 66 ms
5,376 KB
testcase_11 AC 66 ms
5,376 KB
testcase_12 AC 28 ms
5,376 KB
testcase_13 AC 72 ms
5,376 KB
testcase_14 AC 44 ms
5,376 KB
testcase_15 AC 60 ms
5,376 KB
testcase_16 AC 13 ms
5,376 KB
testcase_17 AC 71 ms
5,376 KB
testcase_18 AC 10 ms
5,376 KB
testcase_19 AC 11 ms
5,376 KB
testcase_20 AC 71 ms
5,376 KB
testcase_21 AC 71 ms
5,376 KB
testcase_22 AC 36 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#define overload2(a, b, c, ...) c
#define overload3(a, b, c, d, ...) d
#define overload4(a, b, c, d, e ...) e
#define overload5(a, b, c, d, e, f ...) f
#define overload6(a, b, c, d, e, f, g ...) g
#define fast_io ios::sync_with_stdio(false); cin.tie(nullptr);
#pragma GCC optimize("Ofast,no-stack-protector,unroll-loops,fast-math")
typedef long long ll;
typedef long double ld;
#define chmin(a,b) a = min(a,b);
#define chmax(a,b) a = max(a,b);
#define bit_count(x) __builtin_popcountll(x)
#define leading_zero_count(x) __builtin_clz(x)
#define trailing_zero_count(x) __builtin_ctz(x)
#define gcd(a,b) __gcd(a,b)
#define lcm(a,b) a / gcd(a,b) * b
#define rep(...) overload3(__VA_ARGS__, rrep, rep1)(__VA_ARGS__)
#define rep1(i,n) for(int i = 0 ; i < n ; i++)
#define rrep(i,a,b) for(int i = a ; i < b ; i++)
#define repi(it,S) for(auto it = S.begin() ; it != S.end() ; it++)
#define pt(a) cout << a << endl;
#define print(...) printall(__VA_ARGS__);
#define debug(a) cout << #a << " " << a << endl;
#define all(a) a.begin(), a.end()
#define endl "\n";
#define v1(T,n,a) vector<T>(n,a)
#define v2(T,n,m,a) vector<vector<T>>(n,v1(T,m,a))
#define v3(T,n,m,k,a) vector<vector<vector<T>>>(n,v2(T,m,k,a))
#define v4(T,n,m,k,l,a) vector<vector<vector<vector<T>>>>(n,v3(T,m,k,l,a))
template<typename T,typename U>istream &operator>>(istream&is,pair<T,U>&p){is>>p.first>>p.second;return is;}
template<typename T,typename U>ostream &operator<<(ostream&os,const pair<T,U>&p){os<<p.first<<" "<<p.second;return os;}
template<typename T>istream &operator>>(istream&is,vector<T>&v){for(T &in:v){is>>in;}return is;}
template<typename T>ostream &operator<<(ostream&os,const vector<T>&v){for(auto it=v.begin();it!=v.end();){os<<*it<<((++it)!=v.end()?" ":"");}return os;}
template<typename T>istream &operator>>(istream&is,vector<vector<T>>&v){for(T &in:v){is>>in;}return is;}
template<typename T>ostream &operator<<(ostream&os,const vector<vector<T>>&v){for(auto it=v.begin();it!=v.end();){os<<*it<<((++it)!=v.end()?"\n":"");}return os;}
template<typename T>ostream &operator<<(ostream&os,const set<T>&v){for(auto it=v.begin();it!=v.end();){os<<*it<<((++it)!=v.end()?" ":"");}return os;}
template<typename T>ostream &operator<<(ostream&os,const multiset<T>&v){for(auto it=v.begin();it!=v.end();){os<<*it<<((++it)!=v.end()?" ":"");}return os;}
template<class... Args> void printall(Args... args){for(auto i:initializer_list<common_type_t<Args...>>{args...}) cout<<i<<" ";cout<<endl;}

string S ; // 入力されるめちゃでかい値
string K;
ll dp[20][2][3][21] ; // i桁目にsmallerでkな値

//---------------------------------------------------------------------------//
//                                 debug                                     //
//---------------------------------------------------------------------------//
// cout << "dp[" << i+1 << "][0][" << t << "] = " << dp[i+1][0][t] << endl ; //
// cout << "dp[" << i+1 << "][1][" << t << "] = " << dp[i+1][1][t] << endl ; //
//---------------------------------------------------------------------------//

void solve(){
    memset(dp,0,sizeof(dp));
    cin >> S >> K ;
    int n = S.size() ;
    int k = K.size();

    vector<int> digit, target ; // 各桁の値
    rep(i,n){
        // 各桁の値を入れていく
        digit.push_back(S[i] - '0') ;
    }
    rep(i,k){
        target.push_back(K[i] - '0') ;
    }

    // 初期化を忘れない
    dp[0][0][1][0] = 1 ;
    rep(i,n){
        // Leading 0 がある場合
        { 
            if(i > 0) {
                rrep(x,1,10){
                    if(target[0] == x) dp[i+1][1][1][1] += 1;
                    if(target[0] > x) dp[i+1][1][2][20] += 1;
                }
            }
        }
        // smaller が 0 の場合の処理
        {
            int xxxx , oooo ;
            // 次の桁を見る
            {
                if(i < k && digit[i] < target[i]){
                    dp[i+1][0][2][20] += dp[i][0][1][i] ;
                }
                if(i < k && digit[i] == target[i]){
                    dp[i+1][0][1][i+1] += dp[i][0][1][i] ;
                }
                dp[i+1][0][2][20] += dp[i][0][2][20];
                
                // Leading 0 では上から0桁目に関しては次の桁の値は0としては行けない(03543のような値を作ってしまうため)
                // 1桁目以降は次の桁の値の候補として0~9までどれを入れても良い(1桁目が決まっていれば103843のようになる)
                int fir = 0 ;
                if(i == 0) fir = 1 ;
                rrep(x,fir,digit[i]){
                    if(i < k && target[i] == x) dp[i+1][1][1][i+1] += dp[i][0][1][i];
                    if(i < k && target[i] > x) dp[i+1][1][2][20] += dp[i][0][1][i];
                    dp[i+1][1][2][20] += dp[i][0][2][20];
                }
            }

        }
        // smaller が 1 の場合の処理
        {
            int xxxx, oooo ;
            // smaller == trueの状態から遷移する
            // 自由に遷移して良いので問題によっては対称性を利用することも可能
            rep(x,10) rep(dg,k){
                if(target[dg] == x) dp[i+1][1][1][dg+1] += dp[i][1][1][dg];
                if(target[dg] > x) dp[i+1][1][2][20] += dp[i][1][1][dg];
            }
            dp[i+1][1][2][20] += dp[i][1][2][20] * 10;
        }
    }
    ll res = dp[n][0][2][20] + dp[n][1][2][20];
    rep(i,k) res += dp[n][1][1][i] + dp[n][0][0][i];
    pt(res+1)
}

int main(){
    fast_io
    int t = 1;
    cin >> t;
    rep(i,t) solve();
}
0