結果

問題 No.1645 AB's abs
コンテスト
ユーザー SidewaysOwl
提出日時 2021-08-13 23:21:45
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 110 ms / 2,000 ms
コード長 519 bytes
コンパイル時間 357 ms
コンパイル使用メモリ 82,368 KB
実行使用メモリ 93,312 KB
最終ジャッジ日時 2024-10-03 23:15:17
合計ジャッジ時間 4,495 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 36
権限があれば一括ダウンロードができます

ソースコード

diff # 

n = int(input())
a = list(map(int,input().split()))
dp = [0] * (2 * sum(a) + 1)
dp[0] = 1
lendp = (2 * sum(a) + 1)
mod = 998244353
for i in range(n):
    n_dp = [0] * (lendp)
#     print(dp)
    for j in range(lendp ):
        if dp[j] >= 0:
            n_dp[(j+a[i]) % lendp] = (n_dp[(j+a[i]) % lendp] + dp[j]) % mod 
            n_dp[(j-a[i]) % lendp] = (n_dp[(j-a[i]) % lendp] + dp[j]) % mod
    dp = n_dp[::]
ans = 0
for i in range(sum(a)):
    ans = (ans + (i+1) * dp[(i+1)] + (i+1) * dp[-(i+1)] ) % mod
print(ans)
0