結果
問題 | No.2865 Base 10 Subsets 2 |
ユーザー |
|
提出日時 | 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 |
ソースコード
def main():import sysdata = sys.stdin.read().strip().split()if not data:returnN = 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] = 1for 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 - 1X = K - 1for i in range(L):# i 桁目より下位の桁の選択肢数は suff[i+1]block = suff[i + 1]# 今回選ぶべきグループ番号(0~digits[i]の範囲)を決定d = X // blockans_digits.append(str(d))X %= block # 次の桁に持ち越す残り# 桁列を数値に変換 (先頭の不要な0も int() で除去されます)ans = int("".join(ans_digits))sys.stdout.write(str(ans) + "\n")if __name__ == '__main__':main()