結果

問題 No.3328 岩井ツリーグラフ
コンテスト
ユーザー DeltaStruct
提出日時 2025-05-12 20:10:03
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 96 ms / 2,000 ms
コード長 403 bytes
コンパイル時間 1,936 ms
コンパイル使用メモリ 198,520 KB
実行使用メモリ 7,844 KB
最終ジャッジ日時 2025-05-12 20:10:09
合計ジャッジ時間 4,995 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#include <atcoder/modint>
using mint = atcoder::modint998244353;

int main(){
	int n; cin >> n; vector<int> A(n); mint inv2 = mint(1)/2,inv6 = mint(1)/6; for (int& a:A) cin >> a;
	mint s = accumulate(A.begin(),A.end(),mint(0))+1,r = 0;
	for (int a:A) r += mint(a)*(a+1)*inv2*(s-a)+mint(a)*a*(a+1)*inv2-mint(a)*(a+1)*(2*a+1)*inv6;
	cout << r.val() << endl;
}
0