結果

問題 No.1621 Sequence Inversions
ユーザー convexineq
提出日時 2021-07-23 01:49:34
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 473 ms / 3,000 ms
コード長 832 bytes
コンパイル時間 478 ms
コンパイル使用メモリ 82,560 KB
実行使用メモリ 297,664 KB
最終ジャッジ日時 2024-07-17 23:07:11
合計ジャッジ時間 14,022 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

def polymul(f,g):
    lf = len(f)
    lg = len(g)
    res = [0]*(lf+lg-1)
    for i in range(lf):
        for j in range(lg):
            res[i+j] += f[i]*g[j]
            res[i+j] %= MOD
    return res

def add(r,s):
    if len(r) < len(s):
        r += [0]*(len(s)-len(r))
    for i,si in enumerate(s):
        r[i] += s[i]
        r[i] %= MOD

# dp[i][j] = i 個の隙間に j 個を 
N = 101
MOD = 998244353
dp = [[None]*N for _ in range(N)]
dp[0] = [[1]] + [[0]]*N
for i in range(1,N):
    dp[i][0] = [1]
    for j in range(1,N):
        r = [0]*(i-1) + dp[i][j-1]
        add(r, dp[i-1][j])
        dp[i][j] = r

from itertools import groupby
n,K,*a = map(int,open(0).read().split())
r = [1]
t = 0
for k,v in groupby(sorted(a)):
    c = len(list(v))
    r = polymul(r,dp[t+1][c])
    t += c
print(0 if K >= len(r) else r[K])
0