結果

問題 No.3328 岩井ツリーグラフ
コンテスト
ユーザー 👑 loop0919
提出日時 2025-10-28 22:20:32
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 209 ms / 2,000 ms
コード長 321 bytes
コンパイル時間 323 ms
コンパイル使用メモリ 82,916 KB
実行使用メモリ 102,624 KB
最終ジャッジ日時 2025-10-28 22:20:37
合計ジャッジ時間 5,425 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

MOD = 998244353

X = int(input())
Y = [int(x) for x in input().split()]

sum_Y = sum(Y)
ans = 0

for i in range(X):
    ans += Y[i] ** 2 * (Y[i] - 1) // 2
    ans %= MOD

    ans -= Y[i] * (Y[i] - 1) * (2 * Y[i] - 1) // 6
    ans %= MOD

    ans += Y[i] * (Y[i] + 1) * (sum_Y + 1 - Y[i]) // 2
    ans %= MOD


print(ans)
0