結果

問題 No.876 Range Compress Query
ユーザー 👑 tute7627tute7627
提出日時 2019-09-06 22:21:09
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 299 ms / 2,000 ms
コード長 5,863 bytes
コンパイル時間 1,909 ms
コンパイル使用メモリ 180,164 KB
実行使用メモリ 8,832 KB
最終ジャッジ日時 2024-06-24 19:33:57
合計ジャッジ時間 5,100 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 3 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 3 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 3 ms
5,376 KB
testcase_07 AC 3 ms
5,376 KB
testcase_08 AC 3 ms
5,376 KB
testcase_09 AC 3 ms
5,376 KB
testcase_10 AC 3 ms
5,376 KB
testcase_11 AC 299 ms
8,512 KB
testcase_12 AC 239 ms
8,300 KB
testcase_13 AC 234 ms
8,576 KB
testcase_14 AC 288 ms
8,508 KB
testcase_15 AC 190 ms
8,576 KB
testcase_16 AC 272 ms
8,780 KB
testcase_17 AC 278 ms
8,832 KB
testcase_18 AC 290 ms
8,772 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;

#define ALL(a)  (a).begin(),(a).end()
#define ALLR(a)  (a).rbegin(),(a).rend()
#define spa << " " <<
#define lfs <<fixed<<setprecision(10)<<
#define test cout<<"test"<<endl;
#define fi first
#define se second
#define MP make_pair
#define PB push_back
#define EB emplace_back
#define rep(i,n,m) for(ll i = n; i < (ll)(m); i++)
#define rrep(i,n,m) for(ll i = n - 1; i >= (ll)(m); i--)
typedef long long ll;
typedef long double ld;
const ll MOD = 1e9+7;
//const ll MOD = 998244353;
const ll INF = 1e18;
using P = pair<ll, ll>;
template<typename T>
void chmin(T &a,T b){if(a>b)a=b;}
template<typename T>
void chmax(T &a,T b){if(a<b)a=b;}
void pmod(ll &a,ll b){a=(a+b)%MOD;}
void pmod(ll &a,ll b,ll c){a=(b+c)%MOD;}
void qmod(ll &a,ll b){a=(a*b)%MOD;}
void qmod(ll &a,ll b,ll c){a=(b*c)%MOD;}
ll median(ll a,ll b, ll c){return a+b+c-max({a,b,c})-min({a,b,c});}
void ans1(bool x){if(x) cout<<"Yes"<<endl;else cout<<"No"<<endl;}
void ans2(bool x){if(x) cout<<"YES"<<endl;else cout<<"NO"<<endl;}
void ans3(bool x){if(x) cout<<"Yay!"<<endl;else cout<<":("<<endl;}
template<typename T1,typename T2>
void ans(bool x,T1 y,T2 z){if(x)cout<<y<<endl;else cout<<z<<endl;}  
template<typename T>
void debug(vector<vector<T>>v,ll h,ll w){for(ll i=0;i<h;i++)
{cout<<v[i][0];for(ll j=1;j<w;j++)cout spa v[i][j];cout<<endl;}};
void debug(vector<string>v,ll h,ll w){for(ll i=0;i<h;i++)
{for(ll j=0;j<w;j++)cout<<v[i][j];cout<<endl;}};
template<typename T>
void debug(vector<T>v,ll n){cout<<v[0];
for(ll i=1;i<n;i++)cout spa v[i];cout<<endl;};
template<typename T>
vector<vector<T>>vec(ll x, ll y, T w){
  vector<vector<T>>v(x,vector<T>(y,w));return v;}
ll gcd(ll x,ll y){ll r;while(y!=0&&(r=x%y)!=0){x=y;y=r;}return y==0?x:y;}
template<typename T>
void emp(map<T,ll>&m, T x){m.emplace(x,0).first->second++;}

template<typename T>
struct LazySegmentTree{
  using F = function<T(T,T)>;
  vector<T> data, lazy; 
  ll n,lastlen = 1;
  F func = [](T a, T b){return a<b?a:b;};
  T iden = LLONG_MAX; //identity element
  F f_lazy = [](T a, T b){return a+b;};
  T iden_l = 0;
  F f_trans = [](T a, ll range){return a;};
  LazySegmentTree(vector<T> v){
    n = (ll)v.size();
    while(lastlen < n)lastlen *= 2;
    data.assign(lastlen*2-1,iden);
    lazy.assign(lastlen*2-1,iden_l);
    for(ll i=0;i<n;i++)data[i+lastlen-1] = v[i];
    for(ll i=lastlen-2;i>=0;i--){
      data[i] = func(data[2*i+1], data[2*i+2]);
    }
  }
  void eval(ll k, ll left, ll right){
    if(lazy[k] != iden_l){
      data[k] = f_lazy(data[k],
      f_trans(lazy[k],right-left));
      if(k <= lastlen - 2){
        lazy[2*k+1] = f_lazy(lazy[2*k+1],lazy[k]);
        lazy[2*k+2] = f_lazy(lazy[2*k+2],lazy[k]);
      }
      lazy[k] = iden_l;
    }
  }
  void update(ll a,ll b,T x,ll point=0LL,
  ll left=0LL,ll right=-1LL){
    if(right<0)right=lastlen;
    eval(point,left,right);
    if(b <= left || right <= a);
    else if(a <= left && right <= b ){
      lazy[point] = x;
      eval(point,left,right);
    }
    else{
      update(a,b,x,point*2+1, left, (left+right)/2);
      update(a,b,x,point*2+2, (left+right)/2, right);
      data[point] = func(data[2*point+1], data[2*point+2]);
    }      
  }
  T query(ll a,ll b,ll point=0LL,ll left=0LL,ll right=-1LL){
    if(right<0)right=lastlen;
    T ret = iden;
    eval(point,left,right);
    if(b <= left || right <= a);
    else if(a <= left && right <= b ){
      ret = func(ret,data[point]);
    }
    else{
      T p=query(a,b,point*2+1, left, (left+right)/2);
      T q=query(a,b,point*2+2, (left+right)/2, right);
      data[point] = func(data[2*point+1], data[2*point+2]);
      ret = func(ret,p);
      ret = func(ret,q);
    }      
    return ret;
  }
  void print(){
    ll num=0;
    for(ll i=lastlen;i>0;i/=2)num++;
    ll tmp=1;
    for(ll i=0;i<num;i++,tmp*=2){
      for(ll j=0;j<tmp;j++){
        if(data[tmp-1+j]==iden)cout<<"x"<<" ";
        else cout<<data[tmp-1+j]<<" ";
      };
      cout<<"|";
    }
    cout<<endl;
    tmp=1;
    for(ll i=0;i<num;i++,tmp*=2){
      for(ll j=0;j<tmp;j++){
        if(lazy[tmp-1+j]==iden_l)cout<<"x"<<" ";
        else cout<<lazy[tmp-1+j]<<" ";
      }
      cout<<"|";
    }
    cout<<endl;
  }
};

template<typename T>
struct BIT{
  ll n;
  ll k=1;
  vector<T>data;
  BIT(ll size):n(size){
    data.assign(n,0);
    while(k*2<=n)k*=2;
  }
  void add(ll a,T w){
    for(ll i=a+1;i<=n;i+=i&-i)data[i-1]+=w;
  }
  T sum(ll a){
    T ret = 0;
    for(ll i=a+1;i>0;i-=i&-i)ret+=data[i-1];
    return ret;
  }
  T sum(ll a,ll b){return a==0?sum(b):sum(b)-sum(a-1);}
  ll lower_bound(ll x){
    ll ret=0;    
    for(ll i=k;i>0;i/=2){
      if(ret+i<=n&&data[ret+i-1]<x){
        x-=data[ret+i-1];
        ret+=i;
      }
    }
    return ret;
  }
};

int main(){
  cin.tie(NULL);
  ios_base::sync_with_stdio(false);
  ll res=0,res1=INF,res2=-INF,buf=0;
  bool judge = true;
  ll n,q;cin>>n>>q;
  vector<ll>a(n);
  rep(i,0,n)cin>>a[i];
  LazySegmentTree<ll>segtree(a);
  BIT<ll>bit(n-1);
  rep(i,0,n-1){
    if(a[i]!=a[i+1])bit.add(i,1);
  }
  rep(i,0,q){
    //cout<<bit.sum(0) spa bit.sum(1,1) spa bit.sum(2,2)<<endl;
    ll p,l,r;cin>>p>>l>>r;
    if(p==1){
      ll x;cin>>x;
      segtree.update(l-1,r,x);
      if(l!=1){
        if(bit.sum(l-2,l-2)==1&&
        segtree.query(l-2,l-1)==segtree.query(l-1,l)){
          bit.add(l-2,-1);
        }
        if(bit.sum(l-2,l-2)==0&&
        segtree.query(l-2,l-1)!=segtree.query(l-1,l)){
          bit.add(l-2,1);
        }
      }
      if(r!=n){
        if(bit.sum(r-1,r-1)==1&&
        segtree.query(r-1,r)==segtree.query(r,r+1)){
          bit.add(r-1,-1);
        }
        if(bit.sum(r-1,r-1)==0&&
        segtree.query(r-1,r)!=segtree.query(r,r+1)){
          bit.add(r-1,1);
        }
      }
    }
    else{
      if(l==r)cout<<1<<endl;
      else cout<<bit.sum(l-1,r-2)+1<<endl;
    }
  }
  return 0;
}
0