結果

問題 No.1311 Reverse Permutation Index
ユーザー sotanishy
提出日時 2020-12-08 00:54:01
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 34 ms / 1,500 ms
コード長 512 bytes
コンパイル時間 194 ms
コンパイル使用メモリ 82,108 KB
実行使用メモリ 53,788 KB
最終ジャッジ日時 2024-09-17 14:13:53
合計ジャッジ時間 819 ms
ジャッジサーバーID
(参考情報)
judge2 / judge6
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 6
権限があれば一括ダウンロードができます

ソースコード

diff #

N, S = map(int, input().split())

fact = [1] * (S + 1)
for i in range(1, S + 1):
    fact[i] = i * fact[i - 1]

perm = [0] * S
nums = list(range(1, S + 1))
for i in range(S):
    j = N // fact[S - i - 1]
    N -= fact[S - i - 1] * j
    perm[i] = nums[j]
    nums = nums[:j] + nums[j + 1:]
inv = [0] * S
for i in range(S):
    inv[perm[i] - 1] = i + 1
ans = 0
nums = list(range(1, S + 1))
for i in range(S):
    j = nums.index(inv[i])
    ans += j * fact[S - i - 1]
    nums = nums[:j] + nums[j + 1:]
print(ans)
0