CodeForces - 1454F Array Partition（ST表+二分）

题目链接：https://codeforces.com/problemset/problem/1454/F
借鉴了大神的思路：https://blog.csdn.net/qq_45458915/article/details/110138103

主要是二分这里，x取值从1到n-2遍历，每次二分x+y的值，开始l设为x+1，r设为n-1。
max1[x]表示[1,x]最大值，max2[mid+1]表示[mid+1,n]的最大值。
分类讨论如下：
注意这时候的左边界是固定的，最小值res=query(x+1,mid)
1、res<max1[x]时，说明区间要缩小，r=mid-1
2、res>max1[x]时，说明区间要增大，l=mid+1
3、max1[x]<max2[mid+1]时，说明区间要增大，l=mid+1
4、max1[x]>max2[mid+1]时，说明区间要减小，r=mid-1

然后自己的代码实现了一遍

#include<bits/stdc++.h>
using namespace std;
const int inf=0x3f3f3f3f;
const int maxn=2e5+5;
int Min[maxn][20];
int max1[maxn],max2[maxn];
inline void init(int n){
    for(int j=1;j<=20;j++)
        for(int i=1;i+(1<<j)-1<=n;i++)
            Min[i][j]=min(Min[i][j-1],Min[i+(1<<(j-1))][j-1]);
}
inline int query(int l,int r){
    int k=log2(r-l+1);
    return min(Min[l][k],Min[r-(1<<k)+1][k]);
}
int main(void){
    //freopen("in.txt","r",stdin);
    int t;
    scanf("%d",&t);
    while(t--){
        int n;
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
            scanf("%d",&Min[i][0]);
        init(n);
        int t1=0,t2=0;
        for(int i=1;i<=n;i++){
            t1=max(t1,Min[i][0]);
            t2=max(t2,Min[n-i+1][0]);
            max1[i]=t1;
            max2[n-i+1]=t2;
        }
        bool flag=false;
        int x,y,z;
        for(x=1;x<=n-2;x++){
            int l=x+1,r=n-1,mid=(l+r)>>1;
            while(l<r){
                int res=query(x+1,mid);
                if(res<max1[x]||max1[x]>max2[mid+1]) r=mid-1;//收
                else if(res>max1[x]||max1[x]<max2[mid+1]) l=mid+1;//扩
                else break;
                mid=(l+r)>>1;
            }
            int res=-1;
            if(mid>=x+1)res=query(x+1,mid);
            if(res==max1[x]&&res==max2[mid+1]){
                flag=true; y=mid-x; z=n-x-y; break;
            }
        }
        if(flag)printf("YES\n%d %d %d\n",x,y,z);
        else printf("NO\n");
    }
    return 0;
}