結果
問題 |
No.1000 Point Add and Array Add
|
ユーザー |
|
提出日時 | 2021-10-24 12:17:47 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 694 ms / 2,000 ms |
コード長 | 4,008 bytes |
コンパイル時間 | 6,494 ms |
コンパイル使用メモリ | 203,968 KB |
最終ジャッジ日時 | 2025-01-25 05:08:27 |
ジャッジサーバーID (参考情報) |
judge1 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 22 |
ソースコード
#include<bits/stdc++.h> using namespace std; #define rep(i,n) for(ll i=0;i<n;i++) #define repl(i,l,r) for(ll i=(l);i<(r);i++) #define per(i,n) for(ll i=(n)-1;i>=0;i--) #define perl(i,r,l) for(ll i=r-1;i>=l;i--) #define fi first #define se second #define pb push_back #define ins insert #define pqueue(x) priority_queue<x,vector<x>,greater<x>> #define all(x) (x).begin(),(x).end() #define CST(x) cout<<fixed<<setprecision(x) #define rev(x) reverse(x); using ll=long long; using vl=vector<ll>; using vvl=vector<vector<ll>>; using pl=pair<ll,ll>; using vpl=vector<pl>; using vvpl=vector<vpl>; const ll MOD=1000000007; const ll MOD9=998244353; const int inf=2e9+1; const ll INF=4e18; const ll dy[8]={-1,0,1,0,1,1,-1,-1}; const ll dx[8]={0,-1,0,1,1,-1,1,-1}; template <typename T> inline bool chmax(T &a, T b) { return ((a < b) ? (a = b, true) : (false)); } template <typename T> inline bool chmin(T &a, T b) { return ((a > b) ? (a = b, true) : (false)); } template <typename X, typename M>//faがint lenになってる struct SegTreeLazy {//遅延セグ木 単位元に注意(updateなら選ばれない数、affineなら(1,0)) using FX = function<X(X, X)>; using FA = function<X(X, M, int)>; using FM = function<M(M, M)>; int n; FX fx; FA fa; FM fm; const X ex; const M em; vector<X> dat; vector<M> lazy; SegTreeLazy(int n_, FX fx_, FA fa_, FM fm_, X ex_, M em_) : n(), fx(fx_), fa(fa_), fm(fm_), ex(ex_), em(em_), dat(n_ * 4, ex), lazy(n_ * 4, em) { int x = 1; while (n_ > x) x *= 2; n = x; } void set(int i, X x) { dat[i + n - 1] = x; } void build() { for (int k = n - 2; k >= 0; k--) dat[k] = fx(dat[2 * k + 1], dat[2 * k + 2]); } /* lazy eval */ void eval(int k, int len) { if (lazy[k] == em) return; // 更新するものが無ければ終了 if (k < n - 1) { // 葉でなければ子に伝搬 lazy[k * 2 + 1] = fm(lazy[k * 2 + 1], lazy[k]); lazy[k * 2 + 2] = fm(lazy[k * 2 + 2], lazy[k]); } // 自身を更新 dat[k] = fa(dat[k],lazy[k],len);//fa(dat[k], fp(lazy[k], len)); lazy[k] = em; } void update(int a, int b, M x, int k, int l, int r) { eval(k, r - l); if (a <= l && r <= b) { // 完全に内側の時 lazy[k] = fm(lazy[k], x); eval(k, r - l); } else if (a < r && l < b) { // 一部区間が被る時 update(a, b, x, k * 2 + 1, l, (l + r) / 2); // 左の子 update(a, b, x, k * 2 + 2, (l + r) / 2, r); // 右の子 dat[k] = fx(dat[k * 2 + 1], dat[k * 2 + 2]); } } void update(int a, int b, M x) { update(a, b, x, 0, 0, n); } X query_sub(int a, int b, int k, int l, int r) { eval(k, r - l); if (r <= a || b <= l) { // 完全に外側の時 return ex; } else if (a <= l && r <= b) { // 完全に内側の時 return dat[k]; } else { // 一部区間が被る時 X vl = query_sub(a, b, k * 2 + 1, l, (l + r) / 2); X vr = query_sub(a, b, k * 2 + 2, (l + r) / 2, r); return fx(vl, vr); } } X query(int a, int b) { return query_sub(a, b, 0, 0, n); } X operator[](int i){ return query(i,i+1); } }; struct node{ ll t,l,r; }; int main(){ ll n,q;cin >> n >> q; vector<node> qs; rep(i,n){ ll a;cin >> a; qs.push_back({1,i,a}); } rep(i,q){ char x;cin >> x; ll a,b;cin >> a >> b;a--; if(x=='A')qs.push_back({1,a,b}); else qs.push_back({2,a,b}); } auto fx=[](ll a,ll b){return a+b;}; auto fa=[](ll a,ll b,int len){return a+b*len;}; auto fm=[](ll a,ll b){return a+b;}; SegTreeLazy<ll,ll> st(n,fx,fa,fm,0,0); rev(all(qs)); vl ans(n); for(auto q:qs){ if(q.t==1){ ans[q.l]+=st[q.l]*q.r; } else{ st.update(q.l,q.r,1); } } rep(i,n)cout << ans[i] <<" ";cout << endl; }