結果

問題 No.3328 岩井ツリーグラフ
コンテスト
ユーザー 👑 loop0919
提出日時 2025-05-08 21:32:23
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 310 ms / 2,000 ms
コード長 352 bytes
コンパイル時間 401 ms
コンパイル使用メモリ 82,736 KB
実行使用メモリ 115,328 KB
最終ジャッジ日時 2025-05-11 21:35:13
合計ジャッジ時間 5,296 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

MOD = 998244353

X = int(input())
Y = list(map(int, input().split()))
sum_Y = sum(Y)

ans = 0

for i in range(X):
    ans += Y[i] * (Y[i] + 1) * (2 * Y[i] + 1) * pow(12, -1, MOD)
    ans %= MOD
    ans += -Y[i] * (Y[i] + 1) * pow(4, -1, MOD)
    ans %= MOD
    ans += Y[i] * (Y[i] + 1) * pow(2, -1, MOD) * (sum_Y - Y[i] + 1)
    ans %= MOD

print(ans)
0