yukicoder : No.1435 Mmm......

ライブラリ次第だけど面倒な問題。
https://yukicoder.me/problems/no/1435

問題

整数列Aが与えられる。
以下を満たす区間は何通りあるか答えよ。

• 区間内の最小値2つの和が、最大値以上である。

解法

区間の左端を定めたとき、右端を動かすことを考える。
右端を動かすと最小値はどんどん小さくなり、最大値は大きくなるので、あるところから条件を満たさなくなる。
よって、左端に対し右端を二分探索すればよい。

その際、区間内の最小値2つと最大値を管理できるSegTreeを使うようにしておこう。

int N;
ll A[202020];
int L[202020],R[202020],R2[202020];

template<class V,int NV> class SegTree_Pair {
public:
	vector<pair<V,int> > val;
	static V const def=1<<30;
	pair<V,int> comp(pair<V,int> l,pair<V,int> r){
		
		if(r.first<=l.first) {
			l.second=min(l.first,r.second);
			l.first=r.first;
		}
		else if(r.first<l.second) {
			l.second=r.first;
		}
		
		return l;
	}
	SegTree_Pair(){
		val.resize(NV*2);
		int i;
		FOR(i,NV) val[i+NV]=make_pair(def,def);
		for(i=NV-1;i>=1;i--) val[i]=comp(val[2*i],val[2*i+1]);
	};
	pair<V,int> getval(int x,int y,int l=0,int r=NV,int k=1) {
		if(r<=x || y<=l) return make_pair(def,def);
		if(x<=l && r<=y) return val[k];
		return comp(getval(x,y,l,(l+r)/2,k*2),getval(x,y,(l+r)/2,r,k*2+1));
	}
	void update(int entry, V v) {
		entry += NV;
		val[entry]=make_pair(v,def);
		while(entry>1) entry>>=1, val[entry]=comp(val[entry*2],val[entry*2+1]);
	}
};
SegTree_Pair<int,1<<18> st;
template<class V,int NV> class SegTree_2 {
public:
	vector<V> val;
	static V const def=0;
	V comp(V l,V r){ return max(l,r);};
	
	SegTree_2(){val=vector<V>(NV*2,def);};
	V getval(int x,int y,int l=0,int r=NV,int k=1) { // x<=i<y
		if(r<=x || y<=l) return def;
		if(x<=l && r<=y) return val[k];
		return comp(getval(x,y,l,(l+r)/2,k*2),getval(x,y,(l+r)/2,r,k*2+1));
	}
	void update(int entry, V v) {
		entry += NV;
		val[entry]=comp(v,val[entry]); //上書きかチェック
		while(entry>1) entry>>=1, val[entry]=comp(val[entry*2],val[entry*2+1]);
	}
};
SegTree_2<int,1<<18> st2;


void solve() {
	int i,j,k,l,r,x,y; string s;
	
	cin>>N;
	FOR(i,N) {
		cin>>A[i];
		st.update(i,A[i]);
		st2.update(i,A[i]);
	}
	ll ret=0;
	int L,R;
	FOR(L,N) {
		R=L;
		for(j=18;j>=0;j--) {
			int tmpR=R+(1<<j);
			if(tmpR>=N) continue;
			auto a=st.getval(L,tmpR+1);
			int b=st2.getval(L,tmpR+1);
			if(a.first+a.second>=b) R=tmpR;
		}
		ret+=R-L;
	}
	cout<<ret<<endl;
	
}

まとめ

問題名思いつかなかったのかな…？