結果

問題 No.1645 AB's abs
ユーザー puznekopuzneko
提出日時 2021-10-24 23:03:42
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 115 ms / 2,000 ms
コード長 515 bytes
コンパイル時間 173 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 81,408 KB
最終ジャッジ日時 2024-10-02 05:28:28
合計ジャッジ時間 5,211 ms
ジャッジサーバーID
(参考情報)
judge3 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 36
権限があれば一括ダウンロードができます

ソースコード

diff #

from sys import stdin
n, *a = map(int, stdin.read().split())
p = 998244353
largenum = n * max(a) + 1
maxind = largenum * 2
dp = [[0 for i in range(maxind + 1)] for j in range(n+1)]
dp[0][largenum] = 1

for i in range(n):
    for j in range(maxind + 1):
        if j >= a[i]:
            dp[i+1][j] = dp[i][j-a[i]]
        if j <= maxind - a[i]:
            dp[i+1][j] = (dp[i+1][j] + dp[i][j+a[i]]) % p

ans = 0
for i in range(maxind+1):
    ans = (ans + abs(i-largenum) * dp[n][i] % p) % p
print("{}".format(ans))
0