結果
問題 |
No.3260 岩井スターグラフ
|
ユーザー |
👑 |
提出日時 | 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 |
ソースコード
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)