二分、RMQ/单调栈 Codeforces Round #686 (Div. 3) F题

二分、RMQ/单调栈 Codeforces Round #686 (Div. 3) F题

题意：
将数组恰分成三段，第一段的最大值 = 第二段的最小值 = 第三段的最大值

思路：
看到有博主用单调栈做，看着还挺繁琐的
然后学着用二分,RMQ做了一下

如何二分？ 满足单调性
区间最大值随区间长度增大而增大
区间最小值随区间长度增大而减小

先枚举第一个分界点，然后二分第二个分界点，
如果max1 < min1 || max1 < max2 说明第二个分界点应该往右移，让最小值更小点，第二个最大值更大点
如果max1 > min1 || max1 > max2 说明第二个分界点应该往左移，让最小值更大点，第二个最大值更小点
否则的话，则正好三者都相等

用RMQ求区间最值。

int t,n,m;
int stmax[N][35],stmin[N][35];
 
int get(int l,int r,int flag){
    int k = log2(r - l + 1);
    if(flag) return min(stmin[l][k],stmin[r - (1 << k) + 1][k]);
    else return max(stmax[l][k],stmax[r - (1 << k) + 1][k]);
}
 
void solve(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        int x;
        scanf("%d",&x);
        stmin[i][0] = stmax[i][0] = x;
    }
    for(int j=1;(1<<j) <= n;j++){
        for(int i=1;i+(1<<j)-1<=n;i++){
            stmax[i][j] = max(stmax[i][j-1],stmax[i + (1<<(j-1))][j-1]);
            stmin[i][j] = min(stmin[i][j-1],stmin[i + (1<<(j-1))][j-1]);
        }
    }
    for(int i=1;i<=n-2;i++){
        int max1 = get(1,i,0);
        int l = i+1,r = n-1;
        while(l <= r){
            int mid = l + r >> 1;
            int min1 = get(i+1,mid,1);
            int max2 = get(mid+1,n,0);
            if(max1 < min1 || max1 < max2) l = mid + 1;
            else if(max1 > min1 || max1 > max2) r = mid - 1;
            else{
                puts("YES");
                printf("%d %d %d\n",i,mid - i,n - mid);
                return;
            }
        }
    }
    puts("NO");
}
 
int main(){
    scanf("%d",&t);
    while(t--){
        solve();
    }
    return 0;
}