結果

問題 No.3328 岩井ツリーグラフ
コンテスト
ユーザー DeltaStruct
提出日時 2025-05-08 12:25:30
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 178 ms / 2,000 ms
コード長 354 bytes
コンパイル時間 1,892 ms
コンパイル使用メモリ 196,700 KB
実行使用メモリ 6,272 KB
最終ジャッジ日時 2025-05-11 21:35:16
合計ジャッジ時間 4,820 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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); 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)/2*(s-a)+mint(a)*a*(a+1)/2-mint(a)*(a+1)*(2*a+1)/6;
	cout << r.val() << endl;
}
0