結果

問題 No.1645 AB's abs
ユーザー O2MT
提出日時 2021-08-14 01:45:20
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 153 ms / 2,000 ms
コード長 452 bytes
コンパイル時間 158 ms
コンパイル使用メモリ 82,476 KB
実行使用メモリ 95,104 KB
最終ジャッジ日時 2024-10-04 04:36:13
合計ジャッジ時間 5,546 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 36
権限があれば一括ダウンロードができます

ソースコード

diff #

N = int(input())
A = list(map(int,input().split()))
MOD = 998244353
M = sum(A)
dp = [[0]*(2*M+1) for _ in range(N+1)]
dp[0][M] = 1
for i in range(N):
  for j in range(2*M+1):
    if j-A[i] >= 0:
      if dp[i][j-A[i]] > 0:
        dp[i+1][j] += dp[i][j-A[i]]
    if j+A[i] < 2*M+1:
      if dp[i][j+A[i]] > 0:
        dp[i+1][j] += dp[i][j+A[i]]
    # dp[i+1][j] %= MOD
ans = 0
for j in range(2*M+1):
  ans += dp[N][j]*abs(j-M)
  ans %= MOD

print(ans)
0