【2019牛客暑期多校训练营 第三场 G题】【Removing Stones】【思维】

题目链接：
https://ac.nowcoder.com/acm/contest/883/G
题意：
给一堆石头$a_n$，每次都能从任意两个堆$a_i,a_j$上取一个石头，如果不能找到这样的$a_i,a_j$并且仍有石头留下，就算输。
现在问有几个$1\le l＜r\le n$，使得能够取完所有的石头（如果区间的石头总数是奇数，可以选最少的一堆去掉一个）。
题解：
题目等价于，求满足不包括最大元素的区间元素之和 大于等于 区间最大值 的区间个数。
这里做法是：所有区间的个数减去不符合条件的区间个数。
关键就是求不符合条件的区间个数。
假设每一个$a_i$都是当前区间的最大值来延伸左右区间，并记录相应的区间和，然后统计不符合的区间个数即可。
可以参考代码理解。
代码：

#include<bits/stdc++.h>
#define INF 0x3f3f3f3f
#define sz sizeof
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> Pair;
const int MAX = 3e5 + 10;

int T, N;
ll a[MAX], l[MAX], r[MAX];

int main() {
	ios::sync_with_stdio(0);
	cin >> T;
	while (T--) {
		cin >> N;
		ll ans = 1ll * N * (N + 1) / 2;
		for (int i = 1; i <= N; i++)cin >> a[i];
		for (int i = 1; i <= N; i++) {
			int x = 0, y = 0;
			while (i - x - 1 >= 1 && l[x] + a[i - x - 1] < a[i]) {//向左延伸区间
				x++;
				l[x] = l[x - 1] + a[i - x];//更新左区间的元素和
			}
			while (i + y + 1 <= N && r[y] + a[i + y + 1] < a[i]) {//向右延伸区间
				y++;
				r[y] = r[y - 1] + a[i + y];//更新右区间的元素和
			}
			for (int j = 0; j <= x; j++) {
				while (l[j] + r[y] >= a[i])y--;//因为r[y]是单调增的，所以如果当前符合条件就y--，重新变为不符合条件
				ans -= y + 1;//减去不符合的答案
			}
		}
		cout << ans << endl;
	}
	return 0;
}