#795 Div.2 D. Max GEQ Sum 单调栈

1691D
1800

题意

给出一个序列 $a_1,a_2,...,a_n$，问是否对所有的 $i,j(1\le i\le j\le n,)$，满足 $\max (a_i,...a_j)\ge a_i+...+a_j$。$(n\le 2e5,-10^9\le a_i\le 10^9)$

思路

观察到不等式右侧有 $n^2$ 种取值，而左侧最多只有 $n$ 种取值，于是我们考虑从 $1\to n$ 枚举 $a_i$， 并找到一个区间 $[L,R]$ 满足 $a_i$ 是其中最大的数，在此基础上尽可能最大化 $\sum a[L,R]$。

第一步，找到对应的 $[L_i,R_i]$ 取值范围。也就是分别找到左侧和右侧第一个大于 $a_i$ 的数，可以通过两次单调栈实现。例如计算右侧时，维护一个单调增的栈，从前向后遍历，若当前 $a_i$ 大于栈顶元素对应的值，则栈顶元素右侧第一个大于它的位置是 $i$。另一侧类似。
第二步，最大化 $\sum a[L,R]$。只需维护一个前缀和数组，并找到 $[L,i-1]$ 中的最小值以及 $[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";
}