結果

問題 No.1645 AB's abs
ユーザー player
提出日時 2023-12-23 00:23:49
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 113 ms / 2,000 ms
コード長 571 bytes
コンパイル時間 204 ms
コンパイル使用メモリ 82,388 KB
実行使用メモリ 84,468 KB
最終ジャッジ日時 2024-09-27 11:42:26
合計ジャッジ時間 4,987 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 36
権限があれば一括ダウンロードができます

ソースコード

diff #

n = int(input())
a = list(map(int,input().split()))
mod = 998244353

dp = [[0 for _ in range(2*10**4+2)] for _ in range(n+1)]
dp[0][10**4] = 1
for i in range(n):
    for j in range(2*10**4+2):
        if 0 <= j-a[i] <= 2*10**4+1:
            dp[i+1][j] += dp[i][j-a[i]]
        if 0 <= j+a[i] <= 2*10**4+1:
            dp[i+1][j] += dp[i][j+a[i]]
            
        dp[i+1][j] %= mod
        
ans = 0
st = 10**4
for i in range(2*10**4+2):
    num = (i-st)*dp[n][i]
    # if num != 0:
    #     print(i,num)
    ans += abs((i-st))*dp[n][i]
    ans %= mod
    
print(ans)
0