結果

問題 No.1621 Sequence Inversions
ユーザー ayaoniayaoni
提出日時 2021-07-23 01:01:07
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 2,043 ms / 3,000 ms
コード長 2,714 bytes
コンパイル時間 325 ms
コンパイル使用メモリ 87,128 KB
実行使用メモリ 474,676 KB
最終ジャッジ日時 2023-09-24 21:55:37
合計ジャッジ時間 20,905 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge11
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 96 ms
71,776 KB
testcase_01 AC 96 ms
71,680 KB
testcase_02 AC 147 ms
78,624 KB
testcase_03 AC 125 ms
78,156 KB
testcase_04 AC 732 ms
213,852 KB
testcase_05 AC 96 ms
71,920 KB
testcase_06 AC 118 ms
78,064 KB
testcase_07 AC 146 ms
78,888 KB
testcase_08 AC 838 ms
227,596 KB
testcase_09 AC 1,073 ms
277,980 KB
testcase_10 AC 1,977 ms
474,676 KB
testcase_11 AC 909 ms
242,832 KB
testcase_12 AC 1,998 ms
467,576 KB
testcase_13 AC 2,043 ms
473,044 KB
testcase_14 AC 386 ms
130,648 KB
testcase_15 AC 322 ms
114,584 KB
testcase_16 AC 800 ms
193,136 KB
testcase_17 AC 1,429 ms
305,372 KB
testcase_18 AC 753 ms
185,464 KB
testcase_19 AC 155 ms
79,388 KB
testcase_20 AC 461 ms
127,800 KB
testcase_21 AC 1,574 ms
337,732 KB
testcase_22 AC 451 ms
140,632 KB
testcase_23 AC 1,839 ms
390,764 KB
testcase_24 AC 110 ms
77,176 KB
testcase_25 AC 116 ms
77,556 KB
testcase_26 AC 95 ms
71,748 KB
testcase_27 AC 95 ms
71,692 KB
testcase_28 AC 96 ms
71,852 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
from collections import defaultdict
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,K = MI()
A = LI()
mod = 998244353

count = defaultdict(int)
for a in A:
    count[a] += 1
B = list(count.keys())
B.sort()
M = len(B)

X = [[[0]*(K+1) for _ in range(N+1)] for _ in range(N+1)]
# X[i][j][k] = #{(t1,...,ti)|tiは非負整数,t1+...+ti == j,(i-1)*t1+(i-2)*t2+...+0*ti == k}
X[0][0][0] = 1
for i in range(1,N+1):
    for j in range(N+1):
        for k in range(K+1):
            X[i][j][k] = X[i-1][j][k]
            if j >= 1 and k >= i-1:
                X[i][j][k] += X[i][j-1][k-(i-1)]
                X[i][j][k] %= mod


mod = 998244353
primitive_root = 3  # mod の原始根
roots = [pow(primitive_root,(mod-1) >> i,mod) for i in range(24)]
inv_roots = [pow(r,mod-2,mod) for r in roots]
# roots[i] = 1 の 2**i 乗根、inv_roots[i] = 1 の 2**i 乗根の逆元


def ntt(A,n):
    for i in range(n):
        m = 1 << (n-i-1)
        for start in range(1 << i):
            w = 1
            start *= m*2
            for j in range(m):
                A[start+j],A[start+j+m] = (A[start+j]+A[start+j+m]) % mod,(A[start+j]-A[start+j+m])*w % mod
                w *= roots[n-i]
                w %= mod
    return A


def inv_ntt(A,n):
    for i in range(n):
        m = 1 << i
        for start in range(1 << (n-i-1)):
            w = 1
            start *= m*2
            for j in range(m):
                A[start+j],A[start+j+m] = (A[start+j]+A[start+j+m]*w) % mod,(A[start+j]-A[start+j+m]*w) % mod
                w *= inv_roots[i+1]
                w %= mod
    a = pow(2,n*(mod-2),mod)
    for i in range(1 << n):
        A[i] *= a
        A[i] %= mod
    return A


def convolution(A,B):
    a,b = len(A),len(B)
    deg = a+b-2
    n = deg.bit_length()
    N = 1 << n
    A += [0]*(N-a)  # A の次数を 2冪-1 にする
    B += [0]*(N-b)  # B の次数を 2冪-1 にする
    A = ntt(A,n)
    B = ntt(B,n)
    C = [(A[i]*B[i]) % mod for i in range(N)]
    C = inv_ntt(C,n)
    return C[:deg+1]


dp = [0]*(K+1)
# dp[i][j] = 小さい方からi番目の数からなる数列で、転倒数がjであるものの個数(i省略)
dp[0] = 1
s = count[B[0]]
for i in range(1,M):
    c = count[B[i]]
    XX = X[s+1][c]
    new = convolution(XX,dp)[:K+1]
    dp = new
    s += c

ans = dp[-1]
print(ans)
0