結果
問題 | No.1299 Random Array Score |
ユーザー | zkou |
提出日時 | 2020-11-08 20:39:22 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
AC
|
実行時間 | 83 ms / 2,000 ms |
コード長 | 867 bytes |
コンパイル時間 | 430 ms |
コンパイル使用メモリ | 10,748 KB |
実行使用メモリ | 31,012 KB |
最終ジャッジ日時 | 2023-09-29 21:42:05 |
合計ジャッジ時間 | 4,410 ms |
ジャッジサーバーID (参考情報) |
judge12 / judge15 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 20 ms
8,792 KB |
testcase_01 | AC | 19 ms
8,772 KB |
testcase_02 | AC | 19 ms
8,884 KB |
testcase_03 | AC | 80 ms
31,012 KB |
testcase_04 | AC | 78 ms
30,648 KB |
testcase_05 | AC | 65 ms
25,932 KB |
testcase_06 | AC | 60 ms
24,420 KB |
testcase_07 | AC | 63 ms
24,904 KB |
testcase_08 | AC | 79 ms
30,684 KB |
testcase_09 | AC | 21 ms
9,404 KB |
testcase_10 | AC | 41 ms
17,252 KB |
testcase_11 | AC | 25 ms
10,780 KB |
testcase_12 | AC | 58 ms
22,488 KB |
testcase_13 | AC | 79 ms
30,292 KB |
testcase_14 | AC | 59 ms
23,744 KB |
testcase_15 | AC | 61 ms
24,152 KB |
testcase_16 | AC | 60 ms
23,892 KB |
testcase_17 | AC | 27 ms
11,476 KB |
testcase_18 | AC | 45 ms
17,828 KB |
testcase_19 | AC | 49 ms
20,148 KB |
testcase_20 | AC | 34 ms
14,688 KB |
testcase_21 | AC | 20 ms
9,248 KB |
testcase_22 | AC | 28 ms
12,100 KB |
testcase_23 | AC | 59 ms
23,788 KB |
testcase_24 | AC | 65 ms
25,176 KB |
testcase_25 | AC | 59 ms
22,920 KB |
testcase_26 | AC | 20 ms
9,180 KB |
testcase_27 | AC | 29 ms
12,584 KB |
testcase_28 | AC | 26 ms
11,344 KB |
testcase_29 | AC | 31 ms
13,156 KB |
testcase_30 | AC | 57 ms
22,272 KB |
testcase_31 | AC | 32 ms
13,612 KB |
testcase_32 | AC | 77 ms
29,276 KB |
testcase_33 | AC | 83 ms
30,660 KB |
testcase_34 | AC | 52 ms
12,892 KB |
testcase_35 | AC | 81 ms
30,712 KB |
testcase_36 | AC | 52 ms
12,848 KB |
ソースコード
from itertools import product from random import randint MOD = 998244353 def slow_solve(N, K, As): ret = 0 for choices in product(range(N), repeat=K): A_copy = As.copy() for c in choices: delta = A_copy[c] for i in range(N): A_copy[i] += delta A_copy[i] %= MOD for A in A_copy: ret += A ret %= MOD ret *= pow(N, MOD - 1 - K, MOD) ret %= MOD return ret def solve(N, K, As): return pow(2, K, MOD) * sum(As) % MOD def test(): for _ in range(1000): N = randint(1, 5) K = randint(1, 5) As = [randint(0, 10 ** 9) for _ in range(N)] assert solve(N, K, As) == slow_solve(N, K, As) def main(): N, K = map(int, input().split()) As = list(map(int, input().split())) print(solve(N, K, As)) main()