結果

問題 No.1311 Reverse Permutation Index
ユーザー ayaoniayaoni
提出日時 2020-12-08 00:15:38
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 36 ms / 1,500 ms
コード長 1,041 bytes
コンパイル時間 136 ms
コンパイル使用メモリ 81,984 KB
実行使用メモリ 52,352 KB
最終ジャッジ日時 2024-09-17 14:11:17
合計ジャッジ時間 1,030 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 35 ms
52,224 KB
testcase_01 AC 36 ms
52,224 KB
testcase_02 AC 34 ms
52,224 KB
testcase_03 AC 34 ms
52,352 KB
testcase_04 AC 33 ms
51,840 KB
testcase_05 AC 34 ms
51,712 KB
testcase_06 AC 33 ms
51,712 KB
testcase_07 AC 33 ms
52,224 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys

sys.setrecursionlimit(10**7)
def I(): return int(sys.stdin.readline().rstrip())
def MI(): return map(int,sys.stdin.readline().rstrip().split())
def LI(): return list(map(int,sys.stdin.readline().rstrip().split()))
def LI2(): return list(map(int,sys.stdin.readline().rstrip()))
def S(): return sys.stdin.readline().rstrip()
def LS(): return list(sys.stdin.readline().rstrip().split())
def LS2(): return list(sys.stdin.readline().rstrip())


N,S = MI()

factorial = [1]
for i in range(1,S):
    factorial.append(factorial[-1]*i)

X = []
flag = [0]*(S+1)
for i in range(S):
    r = 1
    while N >= factorial[S-i-1]:
        N -= factorial[S-i-1]
        r += 1
    for i in range(1,S+1):
        if r == 1 and flag[i] == 0:
            flag[i] = 1
            X.append(i)
            break
        elif flag[i] == 0:
            r -= 1

Y = [0]*S
for i in range(S):
    Y[X[i]-1] = i+1

ans = 0
flag = [0]*(S+1)
for i in range(S):
    y = Y[i]
    a = sum(flag[1:y])
    ans += factorial[S-1-i]*(y-1-a)
    flag[y] = 1

print(ans)
0