結果

問題 No.876 Range Compress Query
ユーザー HaarHaar
提出日時 2019-09-07 01:21:37
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 190 ms / 2,000 ms
コード長 4,014 bytes
コンパイル時間 2,603 ms
コンパイル使用メモリ 207,756 KB
実行使用メモリ 6,616 KB
最終ジャッジ日時 2023-09-07 05:55:40
合計ジャッジ時間 5,100 ms
ジャッジサーバーID
(参考情報)
judge12 / judge14
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,380 KB
testcase_01 AC 3 ms
4,376 KB
testcase_02 AC 2 ms
4,376 KB
testcase_03 AC 3 ms
4,380 KB
testcase_04 AC 2 ms
4,380 KB
testcase_05 AC 2 ms
4,380 KB
testcase_06 AC 3 ms
4,376 KB
testcase_07 AC 2 ms
4,380 KB
testcase_08 AC 2 ms
4,380 KB
testcase_09 AC 3 ms
4,380 KB
testcase_10 AC 2 ms
4,376 KB
testcase_11 AC 183 ms
6,412 KB
testcase_12 AC 150 ms
6,616 KB
testcase_13 AC 153 ms
6,484 KB
testcase_14 AC 179 ms
6,416 KB
testcase_15 AC 131 ms
6,532 KB
testcase_16 AC 173 ms
6,472 KB
testcase_17 AC 175 ms
6,448 KB
testcase_18 AC 190 ms
6,452 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#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 LazySegmentTreeRangeAddQuery{
private:
  int size;
  vector<T> vec;
  T e;
  
  inline void propagate(int i){
    if(i<size/2){
      vec[i*2+1] = vec[i] + vec[i*2+1];
      vec[i*2+2] = vec[i] + vec[i*2+2];
      vec[i] = e;
    }
  }
  
  inline void update_aux(int s, int t, int i, int l, int r, const T &x){
    if(r <= s || t <= l) return;
    else if(s <= l && r <= t) vec[i] = vec[i] + x;
    else{
      propagate(i);
      update_aux(s,t,i*2+1,l,(l+r)/2,x);
      update_aux(s,t,i*2+2,(l+r)/2,r,x);
    }
  }
  
  inline void get_aux(int i){
    if(i>0) get_aux((i-1)/2);
    propagate(i);
  }
  
public:
  LazySegmentTreeRangeAddQuery(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 s, int t, const T &x){
    update_aux(s,t,0,0,size/2+1,x);
  }

  inline T get(int i){
    int j=i+size/2;
    get_aux((j-1)/2);
    return vec[j];
  }
};

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;

  LazySegmentTreeRangeAddQuery<LLI> seg1(N,0);
  SegmentTreeRangeSumQuery<int> seg2(N-1,0);

  vector<int> a(N); cin >> a;

  REP(i,N){
    seg1.update(i,i+1,a[i]);
    if(i < N-1 and a[i] != a[i+1]) seg2.update(i,1);
  }

  REP(i,Q){
    int c; cin >> c;
    int l,r; cin >> l >> r;
    --l, --r;

    if(c == 1){
      int x; cin >> x;

      seg1.update(l,r+1,x);

      if(l > 0){
        int t = seg1.get(l-1) != seg1.get(l);
        seg2.update(l-1,t);
      }
      if(r < N-1){
        int t = seg1.get(r) != seg1.get(r+1);
        seg2.update(r,t);
      }
      
    }else{
      int ans = seg2.get(l,r) + 1;
      cout << ans << endl;
    }
  }

  return 0;
}
0