2019牛客暑期多校训练营（第三场） G Removing Stones

传送门

题意：给定N，表示N堆石子，每堆石子数为a[]，问多少个区间，可以满足“石子总和若为偶数，那么可以两两取来自不同堆的石子，直到取完； 如果为奇数，那么排除其中一个，然后可以两两取来自不同堆的石子，直到取完”。等价于求有多少个长度不小于2的连续子数列，使得其中最大元素不大于所有元素和的1/2.

#include<iostream>
#include<cmath>
#include<algorithm>
#include<queue>
#include<cstring>
#include<cstdio>
#include<stack>
#define ll long long
using namespace std;
const int maxn=3e5+10;
ll a[maxn];
ll sum[maxn];
int main()
{
	int t;
	scanf("%d",&t);
	while(t--)
	{
		ll n;
		scanf("%lld",&n);
		ll ans=(n)*(n+1)/2;
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&a[i]);
			sum[i]=sum[i-1]+a[i];
		}
		
		for(int i=1;i<=n;i++)//枚举每个最大值
		{
			int l=i,r=i;
			while(sum[r]-sum[l-1]<2*a[i]&&l>=1) --l;//找到把i为最高点时不满足的左端点
			++l;
			
			for(int j=l;j<=i;j++)
			{
				int x=r;
				while(sum[x]-sum[j-1]<2*a[i]&&x<=n)//找到j为左端点时不满足的右端点
				++x;
				--x;
				ans-=(x-i+1);//答案减去不满足的区间
			}
		}
		printf("%lld\n",ans);
	}
	return 0;
}