結果

問題 No.1697 Deque House
ユーザー chineristACchineristAC
提出日時 2021-08-15 05:30:28
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 663 ms / 3,500 ms
コード長 1,321 bytes
コンパイル時間 1,462 ms
コンパイル使用メモリ 87,020 KB
実行使用メモリ 166,312 KB
最終ジャッジ日時 2023-09-26 15:41:07
合計ジャッジ時間 8,956 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge12
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 70 ms
71,056 KB
testcase_01 AC 69 ms
71,232 KB
testcase_02 AC 69 ms
70,944 KB
testcase_03 AC 71 ms
71,092 KB
testcase_04 AC 71 ms
71,400 KB
testcase_05 AC 70 ms
71,220 KB
testcase_06 AC 76 ms
71,436 KB
testcase_07 AC 100 ms
76,752 KB
testcase_08 AC 92 ms
76,868 KB
testcase_09 AC 658 ms
165,956 KB
testcase_10 AC 287 ms
115,028 KB
testcase_11 AC 325 ms
118,972 KB
testcase_12 AC 270 ms
107,960 KB
testcase_13 AC 209 ms
93,832 KB
testcase_14 AC 259 ms
103,568 KB
testcase_15 AC 400 ms
129,344 KB
testcase_16 AC 366 ms
127,096 KB
testcase_17 AC 282 ms
105,296 KB
testcase_18 AC 256 ms
106,220 KB
testcase_19 AC 290 ms
107,344 KB
testcase_20 AC 647 ms
166,164 KB
testcase_21 AC 650 ms
166,312 KB
testcase_22 AC 663 ms
166,208 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

mod = 998244353

N,K = map(int,input().split())
A = list(map(int,input().split()))

res = 0

f = [[0 for j in range(N+1)] for i in range(K+1)]
for j in range(N+1):
    f[0][j] = 1
t = 1
for i in range(1,K+1):
    t = 2 * t % mod
    f[i][0] = 1
    for j in range(1,N+1):
        f[i][j] = t * f[i][j-1] + f[i-1][j]
        f[i][j] %= mod

def calc_dp(A):
    dp = [[0 for j in range(K+1)] for i in range(N)]
    t = 1
    for j in range(K+1):
        dp[0][j] = t
        t = t * 2 * A[0] % mod
    for i in range(1,N):
        dp[i] = [dp[i-1][j] for j in range(K+1)]
        for j in range(1,K+1):
            dp[i][j] += dp[i][j-1]
            dp[i][j] %= mod
        t0,t1 = 1,1
        for j in range(K+1):
            dp[i][j] = t0 * (f[j][i]*t1+dp[i][j]) % mod
            t0 = 2 * t0 % mod
            t1 = A[i] * t1 % mod
    return dp

dp = calc_dp(A)

c = pow(2,K,mod)
for i in range(1,N):
    rest_way = f[K-1][N-i-1]
    tmp = ((pow(A[i],K,mod) * (c * f[K][i-1] % mod))%mod + dp[i-1][K]) * c % mod
    res += tmp * rest_way % mod
    res %= mod

dp = calc_dp(A[::-1])

for i in range(1,N-1):
    rest_way = c * (c *  f[K][N-2-i] % mod) % mod
    tmp = sum(dp[i-1][k] for k in range(K)) % mod
    res += tmp * rest_way % mod
    res %= mod

i4 = pow(4,mod-2,mod)
res *= pow(i4,K,mod)
print(res % mod)
0