結果

問題 No.3328 岩井ツリーグラフ
コンテスト
ユーザー tau1235
提出日時 2025-11-01 16:58:24
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 157 ms / 2,000 ms
コード長 454 bytes
コンパイル時間 2,850 ms
コンパイル使用メモリ 280,988 KB
実行使用メモリ 7,720 KB
最終ジャッジ日時 2025-11-01 16:58:32
合計ジャッジ時間 5,989 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

int main(){
  using mint=atcoder::modint998244353;
  int x;
  cin>>x;
  vector<mint> y(x);
  vector<mint> s(x);
  mint sum=0;
  mint n=1;
  for (int i=0;i<x;i++){
    int y2;
    cin>>y2;
    y[i]=y2;
    s[i]=y[i]*(y[i]+1)/2;
    sum+=s[i];
    n+=y[i];
  }
  mint ans=0;
  for (int i=0;i<x;i++){
    ans+=n*s[i];
    ans-=(2*y[i]+1)*(y[i]+1)*y[i]/6;
  }
  cout<<ans.val()<<endl;
}
0