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