結果

問題 No.3328 岩井ツリーグラフ
コンテスト
ユーザー Rumain831
提出日時 2026-08-29 09:38:41
言語 C++23(gcc16)
(gcc 16.1.0 + boost 1.92.0)
コンパイル:
g++-16 -O2 -lm -std=c++23 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
AC  
実行時間 59 ms / 2,000 ms
+ 46µs
コード長 2,034 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 2,853 ms
コンパイル使用メモリ 176,472 KB
実行使用メモリ 6,272 KB
最終ジャッジ日時 2026-08-29 09:38:51
合計ジャッジ時間 5,362 ms
ジャッジサーバーID
（参考情報） 		judge2_0 / judge1_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
using ll = long long;
const ll MOD=998244353;

struct mint{
  ll val;
  mint(ll val_=0):val(val_%MOD){
    if(val<0) val+=MOD; 
  }

  mint operator-()const{
    return mint(-val);
  }
  mint& operator+=(const mint& other){
    val+=other.val;
    if(val>=MOD) val-=MOD;
    return *this;
  }
  mint& operator-=(const mint& other){
    val-=other.val;
    if(val<0) val+=MOD;
    return *this;
  }
  mint& operator*=(const mint& other){
    val*=other.val, val%=MOD;
    return *this;
  }
  mint pow(ll n) const{
    mint ans(1);
    mint mul(*this);
    while(n){
      if(n&1) ans*=mul;
      mul*=mul;
      n/=2;
    }
    return ans;
  }
  mint inv() const{
    return pow(MOD-2);
  }
  mint& operator/=(const mint& other){
    return *this*=other.inv();
  }

  friend bool operator==(const mint& lhs, const mint& rhs){
    return lhs.val == rhs.val;
  }
  friend bool operator!=(const mint& lhs, const mint& rhs){
    return lhs.val != rhs.val;
  }
  friend mint operator+(mint lhs, const mint& rhs) {
    lhs+=rhs;
    return lhs;
  }
  friend mint operator-(mint lhs, const mint& rhs) {
    lhs-=rhs;
    return lhs;
  }
  friend mint operator*(mint lhs, const mint& rhs) {
    lhs*=rhs;
    return lhs;
  }
  friend mint operator/(mint lhs, const mint& rhs) {
    lhs/=rhs;
    return lhs;
  }
  friend istream& operator>>(istream& is, mint& m) {
    ll x;
    is >> x;
    m=mint(x);
    return is;
  }
  friend ostream& operator<<(ostream& os, const mint& m) {
    return os << m.val;
  }
};


int main(void){
  int n; cin >> n;
  mint sum=1, inv2=mint(1)/2, inv6=mint(1)/6;
  mint ans=0, arithsum=0;
  auto S1=[&](mint x){return x*(x+1)*inv2;};
  auto S2=[&](mint x){return x*(x+1)*(2*x+1)*inv6;};
  vector<mint> Y(n);
  for(auto&x:Y) cin >> x, sum+=x;
  for(int i=0; i<n; i++){
    mint y=Y[i], rem=sum-y;
    //x=0 ~ y-1
    //(rem+x)*(y-x)=rem*y+(y-rem)x-x^2
    ans+=rem*y*y+(y-rem)*S1(y-1)-S2(y-1);
  }
  cout << ans << endl;
  return 0; 
}
0