結果

問題 No.2865 Base 10 Subsets 2
コンテスト
ユーザー D M
提出日時 2025-02-10 17:07:27
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 42 ms / 2,000 ms
コード長 1,371 bytes
コンパイル時間 2,732 ms
コンパイル使用メモリ 82,196 KB
実行使用メモリ 53,812 KB
最終ジャッジ日時 2025-02-10 17:07:31
合計ジャッジ時間 2,214 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 12
権限があれば一括ダウンロードができます

ソースコード

diff # 

def main():
    import sys
    data = sys.stdin.read().strip().split()
    if not data:
        return
    N = int(data[0])
    K = int(data[1])
    
    # N の十進法表記 (文字列)を取得
    s = str(N)
    L = len(s)
    # 上位桁から順に整数に変換したリスト
    digits = [int(ch) for ch in s]
    
    # 各位置 i について、位置 i+1 から末尾までの「選択肢の総数」をあらかじめ計算
    # suff[i] = (digits[i] + 1) * (digits[i+1] + 1) * ... * (digits[L-1] + 1)
    suff = [1] * (L + 1)
    suff[L] = 1
    for i in range(L - 1, -1, -1):
        suff[i] = suff[i + 1] * (digits[i] + 1)
    
    # K 番目 (1-indexed)に当たる数字を各桁ごとに求める
    # ※各桁の選択肢は 0～digits[i] の (digits[i]+1) 通り
    ans_digits = []
    # 0-indexed に直すため X = K - 1
    X = K - 1
    for i in range(L):
        # i 桁目より下位の桁の選択肢数は suff[i+1]
        block = suff[i + 1]
        # 今回選ぶべきグループ番号（0～digits[i]の範囲）を決定
        d = X // block
        ans_digits.append(str(d))
        X %= block  # 次の桁に持ち越す残り
    # 桁列を数値に変換 (先頭の不要な0も int() で除去されます)
    ans = int("".join(ans_digits))
    sys.stdout.write(str(ans) + "\n")

if __name__ == '__main__':
    main()
0