結果

問題 No.3260 岩井スターグラフ
ユーザー 👑 loop0919
提出日時 2025-05-08 00:31:15
言語 PyPy3
(7.3.15)
結果
RE  
実行時間 -
コード長 657 bytes
コンパイル時間 447 ms
コンパイル使用メモリ 82,732 KB
実行使用メモリ 67,644 KB
最終ジャッジ日時 2025-05-08 00:31:22
合計ジャッジ時間 6,177 ms
ジャッジサーバーID
(参考情報)
judge2 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample RE * 2
other RE * 36
権限があれば一括ダウンロードができます

ソースコード

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)

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

print(ans)
0