【分治 / 玄学尺取】2019牛客暑期多校训练营（第三场） - G - Removing Stones

题目链接https://ac.nowcoder.com/acm/contest/883/G

---

题意

给出n堆石子，每次可以从两堆石子里面各取出一个，如果能去光就获胜。如果总数是奇数，可以先从任意一个最小堆选走一颗石子。问有多少个区间(l,r)，使得只在这个区间内取石子，能获胜。

---

题解一：

分治，复杂度$O(Nlo{g}^{2}N)$
直接上官方题解

---

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=3e5+7;
int t,n,a[N];
int pos[N][33];
ll ans=0,sum[N];
int lg[N];
inline int getk(int l,int r){
    int f=lg[r-l+1];
    if(a[pos[l][f]]>a[pos[r-(1<<f)+1][f]]) return pos[l][f];
    else return pos[r-(1<<f)+1][f];
}
void dfs(int l,int r){
    if(l>=r) return;
    int k=getk(l,r);
    if(r-k<k-l){
        for(int i=r;i>=k;i--){
            if(sum[i]-sum[l-1]<2ll*a[k]) break;
            int lo=l,hi=k,p=0;
            while(lo<=hi){
                int mid=lo+hi>>1;
                if(sum[i]-sum[mid-1]>=2ll*a[k]) lo=mid+1,p=mid;
                else hi=mid-1;
            }
            ans+=p-l+1;
        }
    }
    else{
        for(int i=l;i<=k;i++){
            if(sum[r]-sum[i-1]<2ll*a[k]) break;
            int lo=k,hi=r,p=0;
            while(lo<=hi){
                int mid=lo+hi>>1;
                if(sum[mid]-sum[i-1]>=2ll*a[k]) hi=mid-1,p=mid;
                else lo=mid+1;
            }
            ans+=r-p+1;
        }
    }
    dfs(l,k-1);
    dfs(k+1,r);
}
inline char nc(){
    static char buf[100000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline int _read(){
    char ch=nc();int sum=0;
    while(!(ch>='0'&&ch<='9'))ch=nc();
    while(ch>='0'&&ch<='9')sum=sum*10+ch-48,ch=nc();
    return sum;
}
int main(){
    for(int i=1;i<N;i++) lg[i]=log2(i);
    t=_read();
    while(t--){
        ans=0;
        n=_read();
        for(int i=1;i<=n;i++){
            a[i]=_read();
            sum[i]=sum[i-1]+a[i];
            pos[i][0]=i;
        }
        for(int j=1;(1<<j)<=n;j++){
            for(int i=1;i+(1<<j)-1<=n;i++){
                if(a[pos[i][j-1]]>a[pos[i+(1<<j-1)][j-1]]) pos[i][j]=pos[i][j-1];
                else pos[i][j]=pos[i+(1<<j-1)][j-1];
            }
        }
        dfs(1,n);
        printf("%lld\n",ans);
    }
}

---

题解二

尺取，复杂度玄学
总共有$C_n^2$种区间，减去不符合条件的就是答案。
对于每个点，假设它是最大值，左端点往左移，然后右端点尺取的往右移。

---

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=3e5+7;
int a[N];
int t,n;
int main(){
    scanf("%d",&t);
    while(t--){
        scanf("%d",&n);
        for(int i=1;i<=n;i++) scanf("%d",&a[i]);
        ll ans=1ll*n*(n-1)/2;
        for(int i=1;i<=n;i++){
            ll sum=0;
            int l=i,r=i;
            while(l>1&&sum+a[l-1]<a[i]){
                l--;sum+=a[l];
            }
            if(l<i) ans-=i-l;
            while(l<=i&&r<n){
                while(l<i&&sum+a[r+1]>=a[i]) sum-=a[l],l++;
                if(sum+a[r+1]>=a[i]) break;
                r++;sum+=a[r];
                ans-=i-l+1;
            }
        }
        printf("%lld\n",ans);
    }
}