結果
問題 | No.1733 Sum of Sorted Subarrays |
ユーザー |
👑 ![]() |
提出日時 | 2021-11-07 00:48:14 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 264 ms / 3,000 ms |
コード長 | 3,217 bytes |
コンパイル時間 | 2,253 ms |
コンパイル使用メモリ | 206,880 KB |
最終ジャッジ日時 | 2025-01-25 14:24:46 |
ジャッジサーバーID (参考情報) |
judge10 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 24 |
ソースコード
#include <bits/stdc++.h> #pragma GCC optimize("Ofast") #define _GLIBCXX_DEBUG using namespace std; using std::cout; using std::cin; using std::endl; using ll=long long; using ld=long double; ll ILL=1167167167167167167; const int INF=2100000000; const ll mod=998244353; #define rep(i,a) for (ll i=0;i<a;i++) template<class T> using _pq = priority_queue<T, vector<T>, greater<T>>; template<class T> ll LB(vector<T> &v,T a){return lower_bound(v.begin(),v.end(),a)-v.begin();} template<class T> ll UB(vector<T> &v,T a){return upper_bound(v.begin(),v.end(),a)-v.begin();} template<class T> bool chmin(T &a,const T &b){if(a>b){a=b;return 1;}else return 0;} template<class T> bool chmax(T &a,const T &b){if(a<b){a=b;return 1;}else return 0;} template<class T> void So(vector<T> &v) {sort(v.begin(),v.end());} template<class T> void Sore(vector<T> &v) {sort(v.begin(),v.end(),[](T x,T y){return x>y;});} void yneos(bool a){if(a) cout<<"Yes\n"; else cout<<"No\n";} namespace po167{ template <class T,T (*op)(T,T),T(*e)()> struct segment_tree{ int _n,size; std::vector<T> seg; int ceil_pow2(int a){ int b=1; while(a>b){ b<<=1; } return b; } void update(int k){seg[k]=op(seg[k*2],seg[k*2+1]);}; segment_tree(int n) :_n(n){ size=ceil_pow2(n); seg=std::vector<T>(size*2,e()); } segment_tree(std::vector<T> &p) :_n((int) p.size()){ size=ceil_pow2(_n); seg=std::vector<T>(size*2,e()); for(int i=0;i<_n;i++) seg[i+size]=p[i]; for(int i=size-1;i>0;i--) update(i); } void set(int ind,T val){ assert(0<=ind&&ind<_n); ind+=size; seg[ind]=val; while(ind!=1){ ind>>=1; update(ind); } } T get(int ind){ assert(0<=ind&&ind<_n); return seg[ind+size]; } T query(int l,int r){ assert(0<=l&&l<=r&&r<=_n); T l_val=e(); T r_val=e(); l+=size,r+=size; while(l<r){ if(l&1) l_val=op(l_val,seg[l]),l+=1; if(r&1) r-=1,r_val=op(seg[r],r_val); r>>=1; l>>=1; } return op(l_val,r_val); } template <bool (*f)(T)> int max_right(int l) { return max_right(l, [](T x) { return f(x); }); } template <class F> int max_right(int l, F f) { assert(0<=l&&l<=_n); assert(f(e())); if(f(query(l,_n))) return _n; T val=e(); l+=size; while(true){ while(l%2==0) l>>=1; if(!f(op(val,seg[l]))){ while(l<size){ l*=2; if(f(op(val,seg[l]))){ val=op(val,seg[l]); l++; } } return l-size; } val=op(val,seg[l]); l++; } } }; } using po167::segment_tree; int M=2e5+10; vector<ll> two(M); using F= pair<ll,ll>; F op(F a,F b){ return {a.first+b.first,(a.second*two[b.first]+b.second)%mod}; } F e(){return {0,0};} // rainy ~ 雨に打たれて ~ int main() { ios::sync_with_stdio(false); cin.tie(nullptr); int N; cin>>N; vector<ll> p(N); vector<int> q(N); rep(i,N){ cin>>p[i]; q[i]=i; } two[0]=1; rep(i,M-1){ two[i+1]=(two[i]*2)%mod; } sort(q.begin(),q.end(),[&](int l,int r){ return p[l]<p[r]; }); segment_tree<F,op,e> segl(N),segr(N); rep(i,N) segl.set(i,{0,1}),segr.set(i,{0,1}); ll ans=0; rep(i,N){ int a=q[i]; ll tmp=(segl.query(0,a+1).second*segr.query(0,N-a).second)%mod; ans+=(tmp*p[a])%mod; segl.set(a,{1,2}); segr.set(N-a-1,{1,2}); ans%=mod; } cout<<ans<<"\n"; }