#706166 (C++17) No.1705 Mode of long array

提出ソース

結果

問題	No.1705 Mode of long array
ユーザー	umezo
提出日時	2021-10-08 23:22:38
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	170 ms / 3,000 ms
コード長	2,961 bytes
コンパイル時間	2,274 ms
コンパイル使用メモリ	209,636 KB
実行使用メモリ	7,296 KB
最終ジャッジ日時	2024-07-23 07:20:33
合計ジャッジ時間	8,527 ms
ジャッジサーバーID （参考情報）	judge1 / judge4

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	3 ms 6,812 KB
testcase_01	AC	4 ms 6,944 KB
testcase_02	AC	4 ms 6,940 KB
testcase_03	AC	5 ms 6,940 KB
testcase_04	AC	6 ms 6,940 KB
testcase_05	AC	6 ms 6,944 KB
testcase_06	AC	9 ms 6,940 KB
testcase_07	AC	12 ms 6,940 KB
testcase_08	AC	12 ms 6,944 KB
testcase_09	AC	11 ms 6,940 KB
testcase_10	AC	10 ms 6,944 KB
testcase_11	AC	10 ms 6,940 KB
testcase_12	AC	10 ms 6,940 KB
testcase_13	AC	82 ms 7,004 KB
testcase_14	AC	64 ms 6,940 KB
testcase_15	AC	76 ms 6,940 KB
testcase_16	AC	92 ms 6,944 KB
testcase_17	AC	65 ms 6,944 KB
testcase_18	AC	54 ms 6,940 KB
testcase_19	AC	93 ms 6,940 KB
testcase_20	AC	58 ms 6,944 KB
testcase_21	AC	64 ms 6,940 KB
testcase_22	AC	98 ms 7,040 KB
testcase_23	AC	50 ms 6,940 KB
testcase_24	AC	50 ms 6,940 KB
testcase_25	AC	51 ms 6,940 KB
testcase_26	AC	50 ms 6,940 KB
testcase_27	AC	50 ms 6,944 KB
testcase_28	AC	50 ms 6,944 KB
testcase_29	AC	50 ms 6,940 KB
testcase_30	AC	50 ms 6,944 KB
testcase_31	AC	51 ms 6,940 KB
testcase_32	AC	51 ms 6,940 KB
testcase_33	AC	73 ms 7,036 KB
testcase_34	AC	74 ms 7,176 KB
testcase_35	AC	74 ms 7,168 KB
testcase_36	AC	74 ms 7,164 KB
testcase_37	AC	74 ms 7,040 KB
testcase_38	AC	75 ms 6,980 KB
testcase_39	AC	75 ms 7,168 KB
testcase_40	AC	74 ms 7,100 KB
testcase_41	AC	75 ms 7,164 KB
testcase_42	AC	73 ms 7,044 KB
testcase_43	AC	170 ms 7,152 KB
testcase_44	AC	170 ms 7,168 KB
testcase_45	AC	167 ms 7,268 KB
testcase_46	AC	167 ms 7,168 KB
testcase_47	AC	164 ms 7,164 KB
testcase_48	AC	167 ms 7,168 KB
testcase_49	AC	98 ms 7,212 KB
testcase_50	AC	98 ms 7,296 KB
testcase_51	AC	98 ms 7,168 KB
testcase_52	AC	96 ms 7,088 KB
testcase_53	AC	97 ms 7,168 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#define rep(i,n) for(int i=0;i<(int)(n);i++)
#define ALL(v) v.begin(),v.end()
typedef long long ll;

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

const int INF=2147483647;

template <typename X>
struct SegTree{
  using FX=function<X(X,X)>;// X•X -> X となる関数の型
  int n;
  FX fx;
  const X ex;
  vector<X> dat;
  SegTree(int n_,FX fx_,X ex_) : n(),fx(fx_),ex(ex_),dat(n_*4,ex_){
    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]);
  }
  void update(int i,X x){
    i += n-1;
    dat[i]=x;
    while(i>0){
      i=(i-1)/2; // parent
      dat[i]=fx(dat[i*2+1],dat[i*2+2]);
    }
  }
  X get(int i){return dat[i+n-1];}
  X all(){return dat[0];}
  X query(int a,int b){ return query_sub(a,b,0,0,n);}
  X query_sub(int a,int b,int k,int l,int r){
    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);
    }
  }
  int find_rightest(int a,int b,X x){ return find_rightest_sub(a,b,x,0,0,n);}
  int find_leftest(int a,int b,X x){ return find_leftest_sub(a,b,x,0,0,n);}
  int find_rightest_sub(int a,int b,X x,int k,int l,int r){
    if(dat[k]<x || r<=a || b<=l){  // 自分の値がxより大きい or [a,b)が[l,r)の範囲外ならreturn a-1
      return a-1;
    }else if (k>=n-1){  // 自分が葉ならその位置をreturn
      return (k-(n-1));
    }else{
      int vr=find_rightest_sub(a,b,x,2*k+2,(l+r)/2,r);
      if(vr!=a-1){  // 右の部分木を見て a-1 以外ならreturn
        return vr;
      }else{  // 左の部分木を見て値をreturn
        return find_rightest_sub(a,b,x,2*k+1,l,(l+r)/2);
      }
    }
  }
  int find_leftest_sub(int a,int b,X x,int k,int l,int r){
    if(dat[k]<x || r<=a || b<=l){  // 自分の値がxより大きい or [a,b)が[l,r)の範囲外ならreturn b
      return b;
    }else if (k>=n-1){  // 自分が葉ならその位置をreturn
      return (k-(n-1));
    }else{
      int vl=find_leftest_sub(a,b,x,2*k+1,l,(l+r)/2);
      if(vl != b){  // 左の部分木を見て b 以外ならreturn
        return vl;
      }else{  // 右の部分木を見て値をreturn
        return find_leftest_sub(a,b,x,2*k+2,(l+r)/2,r);
      }
    }
  }
};

int main(){
  ios::sync_with_stdio(false);
  std::cin.tie(nullptr);

  ll n,m;
  cin>>n>>m;
  
  vector<ll> A(m);
  rep(i,m) cin>>A[i];
  
  using X=ll;
  auto fx=[](X x1,X x2)->X{ return max(x1,x2);};
  X ex=0;
  SegTree<X> seg(100010,fx,ex);
  
  rep(i,m) seg.set(i+1,A[i]);
  seg.build();
  
  int q;
  cin>>q;
  while(q--){
    ll t,x,y;
    cin>>t>>x>>y;
    if(t==1) seg.update(x,seg.query(x,x+1)+y);
    else if(t==2) seg.update(x,seg.query(x,x+1)-y);
    else{
      ll ma=seg.all();
      cout<<seg.find_rightest(1,m+1,ma)<<endl;
    }
  }
      
  return 0;
}

yukicoder

結果

テストケース

ソースコード