結果

問題 No.1435 Mmm......
コンテスト
ユーザー umezo
提出日時 2021-03-20 01:20:44
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 3,075 bytes
コンパイル時間 3,697 ms
コンパイル使用メモリ 210,944 KB
最終ジャッジ日時 2025-01-19 20:06:29
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 3 TLE * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

#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=1e9+7;

template <typename X>
struct SegTree{
  using FX=function<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;
      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(){
  int n;
  cin>>n;
  vector<int> A(n);
  rep(i,n) cin>>A[i];
  
  using X=int;
  auto fx=[](X x1,X x2)->X{ return min(x1,x2);};
  X ex=INF;
  SegTree<X> seg(n,fx,ex);
  
  auto fx1=[](X x1,X x2)->X{ return max(x1,x2);};
  X ex1=0;
  SegTree<X> seg1(n,fx1,ex1);
  
  rep(i,n){
    seg.set(i,A[i]);
    seg1.set(i,A[i]);
  }
  seg.build();
  seg1.build();
  
  ll ans=0;
  rep(i,n-1){
    int ok=i+1,ng=n;
    while(ng-ok>1){
      int mid=(ng+ok)/2;
      int M=seg1.query(i,mid+1);
      int m1=seg.query(i,mid+1);
      int l=seg.find_leftest(i,mid+1,m1);
      int m2=min(seg.query(i,l),seg.query(l+1,mid+1));
      if(M<=m1+m2) ok=mid;
      else ng=mid;
    }
    ans+=ok-i;
  }
  cout<<ans<<endl;
    
  return 0;
}
0