yukicoder No.362 門松ナンバー

問題

No.362 門松ナンバー - yukicoder

解法

calc(x) = (x 以下の門松ナンバーがいくつあるか) というのがある程度高速に求められたとすると, x に関する二分探索をやれば良いです。

ということで, calc 関数が欲しいですが, これは桁dp をやれば良いです。

dp[桁][100で割った余り][xより小さいか][有効な数字の数] = 門松ナンバーの数
有効な数字の数, というのは例えば最初 3 文字が 000 でもその後に 000153 と続けばこれは有効な門松ナンバーなのでそれを考慮するためのものです。

bool isKadomatsu(const vector<int>& A) {
    if (A[0] == A[1] || A[0] == A[2] || A[1] == A[2]) return false;
    if (A[0] > A[1] && A[1] < A[2]) return true;
    if (A[0] < A[1] && A[1] > A[2]) return true;
    return false;
}

// keta, %100, small, renzoku
ll dp[17][100][2][4];
// x 以下の門松ナンバーの数を数える
ll calc(ll x) {
    memset(dp, 0, sizeof(dp));
    string s = to_string(x);
    dp[1][0][1][0] = 1;
    for (int i = 1; i < 10; i++) {
        int num = (s[0]-'0');
        if (i > num) break;
        dp[1][i][i < num][1] = 1;
    }
    int n = s.size();
    for (int keta = 1; keta < n; keta++) for (int q = 0; q < 100; q++) {
        for (int small = 0; small < 2; small++) for (int renzoku = 0; renzoku < 4; renzoku++) {
            for (int num = 0; num < 10; num++) {
                if (!small && num > (s[keta]-'0')) break;
                if (renzoku >= 2) {
                    vi A(3);
                    A[0] = (q/10), A[1] = q%10, A[2] = num;
                    if (!isKadomatsu(A)) continue;
                }
                int nrenzoku = renzoku;
                if (renzoku != 0 || num != 0) nrenzoku = min(3, nrenzoku+1);
                int nsmall = small;
                nsmall |= (num < s[keta]-'0');
                int nq = (q*10+num)%100;
                dp[keta+1][nq][nsmall][nrenzoku] += dp[keta][q][small][renzoku];
            }
        }
    }
    ll ret = 0;
    for (int i = 0; i < 100; i++) for (int j = 0; j < 2; j++) {
        ret += dp[n][i][j][3];
    }
    return ret;
}

int main() {
    cin.tie(0);
    ios::sync_with_stdio(false);
    int T;
    cin >> T;
    while (T--) {
        ll K;
        cin >> K;
        ll low = 100, high = 1e15;
        while (high-low > 1) {
            const ll med = (high+low)/2;
            if (calc(med) >= K) high = med;
            else low = med;
        }
        cout << high << endl;
    }
    return 0;
}