結果

問題 No.2950 Max Min Product
ユーザー askr58
提出日時 2024-10-25 21:59:37
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,441 bytes
コンパイル時間 5,936 ms
コンパイル使用メモリ 319,012 KB
実行使用メモリ 19,988 KB
最終ジャッジ日時 2024-10-25 21:59:56
合計ジャッジ時間 10,782 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other TLE * 1 -- * 35
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
using ll=long long;
using mint=modint998244353;
using pi=pair<int,int>;
pi op1(pi x,pi y){
    return min(x,y);
}
pi op2(pi x,pi y){
    return max(x,y);
}
pi e1(){
    return make_pair((int)2e9,-1);
}
pi e2(){
    return make_pair(0,-1);
}
int main()
{
    int n;
    cin>>n;
    vector<pi> a(n);
    vector<mint> b(n);
    for(int i=0;i<n;i++){
        cin>>a[i].first;
        a[i].second=i;
        b[i]=a[i].first;
    }

    
    segtree<pi,op1,e1> st_min(a);
    segtree<pi,op2,e2> st_max(a);
    mint ans=0;
    auto sub_rec=[&](auto rec,int l,int r,int pivot)->void{
        int ind=st_min.prod(l,r).second;
        if(pivot==ind){
            ans+=(b[pivot]*b[pivot]*(pivot-l+1))*(r-pivot);
            return;
        }else if(pivot<ind){
            ans+=(b[pivot]*b[ind]*(pivot-l+1))*(r-ind);
            rec(rec,l,ind,pivot);
            return;
        }else{
            ans+=(b[pivot]*b[ind]*(ind-l+1))*(r-pivot);
            rec(rec,ind+1,r,pivot);
            return;
        }
    };
    auto rec=[&](auto rec,int l,int r,int pivot)->void{
        if(r==l){
            return;
        }
        sub_rec(sub_rec,l,r,pivot);
        rec(rec,l,pivot,st_max.prod(l,pivot).second);
        rec(rec,pivot+1,r,st_max.prod(pivot+1,r).second);
    };
    rec(rec,0,n,st_max.prod(0,n).second);
    cout<<ans.val()<<endl;
    return 0;
}
0