結果

問題 No.1311 Reverse Permutation Index
コンテスト
ユーザー shotoyoo
提出日時 2020-12-08 00:24:26
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 39 ms / 1,500 ms
コード長 1,041 bytes
コンパイル時間 189 ms
コンパイル使用メモリ 82,312 KB
実行使用メモリ 53,612 KB
最終ジャッジ日時 2024-09-17 14:11:43
合計ジャッジ時間 975 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 6
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = lambda : sys.stdin.readline().rstrip()
sys.setrecursionlimit(max(1000, 10**9))
write = lambda x: sys.stdout.write(x+"\n")

n,s = map(int, input().split())
g = [1]
v = 1
for i in range(1,s+2):
    v *= i
    g.append(v)
def ps(n,v):
    """長さnの順列で、辞書順v番目
    """
    seen = [False]*n
    ans = []
    for i in range(n):
        num = v//g[n-i-1]
        count = 0
        for j in range(n):
            if seen[j]:
                continue
            if count==num:
                break
            if not seen[j]:
                count += 1
        ans.append(j)
        seen[j] = True
        v = v%g[n-i-1]
    return ans
def index(ps):
    """順列psの辞書順
    """
    n = len(ps)
    v = 0
    seen = [False] * n
    for i in range(n):
        num = 0
        for j in range(ps[i]):
            if not seen[j]:
                num += 1
        v += num * g[n-i-1]
        seen[ps[i]] = True
    return v
p = ps(s,n)
pp = [None]*s
for i,v in enumerate(p):
    pp[v] = i
print(index(pp))
0