1691D
1800
题意
给出一个序列 a 1 , a 2 , . . . , a n a_1, a_2,...,a_n a1,a2,...,an,问是否对所有的 i , j ( 1 ≤ i ≤ j ≤ n , ) i,j(1\leq i\leq j\leq n,) i,j(1≤i≤j≤n,),满足 max ( a i , . . . a j ) ≥ a i + . . . + a j \max(a_i,...a_j)\geq a_i+...+a_j max(ai,...aj)≥ai+...+aj。 ( n ≤ 2 e 5 , − 1 0 9 ≤ a i ≤ 1 0 9 ) (n\leq 2e5,-10^9\leq a_i\leq 10^9) (n≤2e5,−109≤ai≤109)
思路
观察到不等式右侧有 n 2 n^2 n2 种取值,而左侧最多只有 n n n 种取值,于是我们考虑从 1 → n 1\to n 1→n 枚举 a i a_i ai, 并找到一个区间 [ L , R ] [L,R] [L,R] 满足 a i a_i ai 是其中最大的数,在此基础上尽可能最大化 ∑ a [ L , R ] \sum a[L,R] ∑a[L,R]。
第一步,找到对应的
[
L
i
,
R
i
]
[L_i,R_i]
[Li,Ri] 取值范围。也就是分别找到左侧和右侧第一个大于
a
i
a_i
ai 的数,可以通过两次单调栈实现。例如计算右侧时,维护一个单调增的栈,从前向后遍历,若当前
a
i
a_i
ai 大于栈顶元素对应的值,则栈顶元素右侧第一个大于它的位置是
i
i
i。另一侧类似。
第二步,最大化
∑
a
[
L
,
R
]
\sum a[L,R]
∑a[L,R]。只需维护一个前缀和数组,并找到
[
L
,
i
−
1
]
[L,i-1]
[L,i−1] 中的最小值以及
[
i
,
R
]
[i, R]
[i,R] 中的最大值相减即可。可以用线段树或者ST表,注意别忘了前缀和数组下标从0开始。
代码
int n, a[maxn], L[maxn], R[maxn];
int pre[maxn];
struct segment_tree {
int l, r;
int minn, maxx;
#define l(x) tree[x].l
#define r(x) tree[x].r
#define mn(x) tree[x].minn
#define mx(x) tree[x].maxx
}tree[maxn << 2];
void pushup(int p) {
mn(p) = min(mn(p<<1), mn(p<<1|1));
mx(p) = max(mx(p << 1), mx(p << 1 | 1));
}
void build(int p, int l, int r) {
l(p) = l, r(p) = r;
if(l == r) {
mn(p) = mx(p) = pre[l];
return;
}
int mid = (l + r) >> 1;
build(p << 1, l, mid);
build(p << 1 | 1, mid + 1, r);
pushup(p);
}
int query_min(int p, int l, int r) {
if(l(p) >= l && r(p) <= r) {
return mn(p);
}
int mid = (l(p) + r(p)) >> 1;
int ans = (1ll<<62);
if(l <= mid) ans = min(ans, query_min(p << 1, l, r));
if(r > mid) ans = min(ans, query_min(p << 1 | 1, l, r));
return ans;
}
int query_max(int p, int l, int r) {
if(l(p) >= l && r(p) <= r) {
return mx(p);
}
int mid = (l(p) + r(p)) >> 1;
int ans = -(1ll<<62);
if(l <= mid) ans = max(ans, query_max(p << 1, l, r));
if(r > mid) ans = max(ans, query_max(p << 1 | 1, l, r));
return ans;
}
void solve() {
int last = 0;
cin >> n;
pre[0] = 0;
for(int i = 1; i <= n; i++) {
cin >> a[i];
pre[i] = pre[i-1] + a[i];
}
a[n+1] = a[0] = INF;
stack<int> s;
for(int i = 1; i <= n + 1; i++) {
while(s.size() && a[s.top()] < a[i]) {
R[s.top()] = i;
s.pop();
}
s.push(i);
}
while(s.size()) s.pop();
for(int i = n; i >= 0; i--) {
while(s.size() && a[s.top()] < a[i]) {
L[s.top()] = i;
s.pop();
}
s.push(i);
}
while(s.size()) s.pop();
build(1, 1, n);
for(int i = 1; i <= n; i++) {
int mnn = query_min(1, L[i], i-1);
if(L[i] == 0) mnn = min(mnn, 0LL);
int mxx = query_max(1, i, R[i] - 1);
if(mxx - mnn > a[i]) {
cout << "NO\n";
return;
}
}
cout << "YES\n";
}