結果

問題 No.3328 岩井ツリーグラフ
コンテスト
ユーザー 👑 loop0919
提出日時 2025-05-08 00:43:58
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 672 bytes
コンパイル時間 438 ms
コンパイル使用メモリ 82,116 KB
実行使用メモリ 111,724 KB
最終ジャッジ日時 2025-05-11 21:35:11
合計ジャッジ時間 5,508 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 3 TLE * 1 -- * 17
権限があれば一括ダウンロードができます

ソースコード

diff # 

MOD = 998244353

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

def count(n):
    val = n * (n + 1) % MOD * (2 * n + 1) % MOD * pow(6, -1, MOD) % MOD
    val += n * (n + 1) % MOD * pow(2, -1, MOD) % MOD
    val *= pow(2, -1, MOD)
    return val % MOD

def sigma(n):
    return n * (n + 1) % MOD * pow(2, -1, MOD) % MOD

ans = 0
for i in range(X):
    for j in range(i + 1, X):
        ans += count(Y[i] + Y[j])
        ans -= count(Y[i] - 1)
        ans -= count(Y[j] - 1)
        ans -= sigma(Y[i]) + sigma(Y[j])
        ans %= MOD

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

for i in range(X):
    ans += sigma(Y[i])
    ans %= MOD

print(ans)
0