2019牛客暑期多校训练营（第三场）G: Removing Stones（启发式分治）

题意：给定N，表示N堆石子，每堆石子数为a[]，问多少个区间，可以满足“石子总和若为偶数，那么可以两两取来自不同堆的石子，直到取完； 如果为奇数，那么排除其中一个，然后可以两两取来自不同堆的石子，直到取完”。

思路：结论是，如果一个区间的区间和大于等于区间最大值的两倍，则这个区间合法。 考虑分治，我们首先找到区间最大值(为了不重复统计，多个最大值时，统一取最左边的，这个可以ST表示实现)，然后考虑跨越这个位置的合法区间个数。枚举一端，另外一段二分即可。

由于分治的性质，我们每次的复杂度要倾向于小的那边，即是一个启发式合并的逆过程，所以启发式分治复杂度是O(NlogN)的，加上二分，这个做法的复杂度是O(Nlog^2N)。

可以参考差不多的题：https://www.cnblogs.com/hua-dong/p/11171241.html。所以我感觉我遇到的原题还挺多的。

#include<bits/stdc++.h>
#define ll long long
#define rep(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
const int maxn=;
int lg[maxn],a[maxn],dp[maxn][],N;
ll sum[maxn],sum2[maxn],ans;
void RMQ()
{
    rep(i,,N) dp[i][]=i;
    for(int i=;(<<i)<=N;i++){
        for(int j=;j+(<<i)-<=N;j++){
           dp[j][i]=a[dp[j][i-]]>=a[dp[j+(<<(i-))][i-]]?
             dp[j][i-]:dp[j+(<<(i-))][i-];
        }
    }
}
void solve(int L,int R)
{
    if(L>=R) return ;
    int k=lg[R-L+];
    int Mid=(a[dp[L][k]]>=a[dp[R-(<<k)+][k]]?
      dp[L][k]:dp[R-(<<k)+][k]);
    if(Mid-L<R-Mid){
        rep(i,L,Mid) {
            int pos=lower_bound(sum+Mid,sum+R+,sum[i-]+2LL*a[Mid])-sum;
            ans+=R-pos+;
        }
    }
    else {
        rep(i,Mid,R) {
            int pos=lower_bound(sum2+N-Mid+,sum2+N-L+,sum2[N-i]+2LL*a[Mid])-sum2;
            ans+=N+-L-pos+;
        }
    }
    solve(L,Mid-); solve(Mid+,R);
}
int main()
{
    lg[]=-;
    rep(i,,maxn-) lg[i]=lg[i>>]+;
    int T;
    scanf("%d",&T);
    while(T--){
        scanf("%d",&N); ans=;
        rep(i,,N)
          scanf("%d",&a[i]),sum[i]=sum[i-]+a[i];
        rep(i,,N) sum2[i]=sum2[i-]+a[N+-i];
        RMQ();
        solve(,N);
        printf("%lld\n",ans);
    }
    return ;
}