結果
問題 | No.877 Range ReLU Query |
ユーザー |
|
提出日時 | 2019-09-07 01:57:09 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 257 ms / 2,000 ms |
コード長 | 3,154 bytes |
コンパイル時間 | 2,572 ms |
コンパイル使用メモリ | 219,644 KB |
最終ジャッジ日時 | 2025-01-07 17:07:37 |
ジャッジサーバーID (参考情報) |
judge5 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 20 |
ソースコード
#include <bits/stdc++.h> #define LLI long long int #define FOR(v, a, b) for(LLI v = (a); v < (b); ++v) #define FORE(v, a, b) for(LLI v = (a); v <= (b); ++v) #define REP(v, n) FOR(v, 0, n) #define REPE(v, n) FORE(v, 0, n) #define REV(v, a, b) for(LLI v = (a); v >= (b); --v) #define ALL(x) (x).begin(), (x).end() #define RALL(x) (x).rbegin(), (x).rend() #define ITR(it, c) for(auto it = (c).begin(); it != (c).end(); ++it) #define RITR(it, c) for(auto it = (c).rbegin(); it != (c).rend(); ++it) #define EXIST(c,x) ((c).find(x) != (c).end()) #define fst first #define snd second #define popcount __builtin_popcount #define UNIQ(v) (v).erase(unique(ALL(v)), (v).end()) #define bit(i) (1LL<<(i)) #ifdef DEBUG #include <misc/C++/Debug.cpp> #else #define dump(...) ((void)0) #endif #define gcd __gcd using namespace std; template <class T> constexpr T lcm(T m, T n){return m/gcd(m,n)*n;} template <typename I> void join(ostream &ost, I s, I t, string d=" "){for(auto i=s; i!=t; ++i){if(i!=s)ost<<d; ost<<*i;}ost<<endl;} template <typename T> istream& operator>>(istream &is, vector<T> &v){for(auto &a : v) is >> a; return is;} template <typename T, typename U> bool chmin(T &a, const U &b){return (a>b ? a=b, true : false);} template <typename T, typename U> bool chmax(T &a, const U &b){return (a<b ? a=b, true : false);} template <typename T, size_t N, typename U> void fill_array(T (&a)[N], const U &v){fill((U*)a, (U*)(a+N), v);} struct Init{ Init(){ cin.tie(0); ios::sync_with_stdio(false); } }init; template <typename T> class SegmentTreeRangeSumQuery{ private: int size; vector<T> vec; T e; inline T aux(int x, int y, int i, int l, int r){ if(r<=x || y<=l) return e; else if(x<=l && r<=y) return vec[i]; else return aux(x,y,i*2+1,l,(l+r)/2) + aux(x,y,i*2+2,(l+r)/2,r); }; public: SegmentTreeRangeSumQuery(int n, const T &e): e(e){ size = 1; while(size < n) size *= 2; size = size*2-1; vec = vector<T>(size, e); } inline void update(int i, const T &x){ int j = i+(size+1)/2-1; vec[j] = x; --j; while(j>=0){ vec[j/2] = vec[(j/2)*2+1] + vec[(j/2)*2+2]; (j /= 2) -= 1; } } inline T at(int i){ return vec[(size+1)/2 + i - 1]; } inline T get(int x, int y){ // [x,y) return aux(x,y,0,0,(size+1)/2); } }; int main(){ int N,Q; cin >> N >> Q; vector<int> a(N); cin >> a; SegmentTreeRangeSumQuery<LLI> seg1(N,0), seg2(N,0); REP(i,N){ seg1.update(i,a[i]); seg2.update(i,1); } vector<LLI> ans(Q); vector<tuple<int,int,int,int>> qs(Q); REP(i,Q){ int c,l,r,x; cin >> c >> l >> r >> x; --l, --r; qs[i] = make_tuple(x,l,r,i); } sort(ALL(qs)); vector<pair<int,int>> b(N); REP(i,N){ b[i] = make_pair(a[i], i); } sort(RALL(b)); REP(i,Q){ int x,l,r,j; tie(x,l,r,j) = qs[i]; while(b.size() and b.back().fst <= x){ seg1.update(b.back().snd, 0); seg2.update(b.back().snd, 0); b.pop_back(); } LLI t = seg1.get(l,r+1) - seg2.get(l,r+1) * x; ans[j] = t; } REP(i,Q){ cout << ans[i] << endl; } return 0; }