結果
問題 | No.1645 AB's abs |
ユーザー | puzneko |
提出日時 | 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 |
ソースコード
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))