牛客小白月赛16 小阳买水果 线段树

小阳买水果

从大佬处学习的

求出ai的前缀和，用线段树维护前缀和最大值，

从 0 遍历到 n ，在线段树中找到大于sum[i]的最远的一个编号，每次先往右找，保证找到的值是最远。然后取最远距离 - i 的最大值。

#include<bits/stdc++.h>
using namespace std;
#define Mid ((l+r)>>1)
#define Left rt<<1,l,Mid
#define Right rt<<1|1,Mid+1,r
typedef long long ll;
const int N=2e6+10;

ll sum[N];
ll tree[4*N];
void bulid(int rt, int l, int r){
	if(l==r){
		tree[rt]=sum[l];
		return ;
	}
	bulid(Left);
	bulid(Right);
	tree[rt] = max(tree[rt<<1], tree[rt<<1|1]);
}

int query(int rt, int l, int r, ll value){
	if(l == r){
		return l;
	}
	if(tree[rt<<1|1] > value) return query(Right,value);
	if(tree[rt<<1] > value) return query(Left,value);
	return 0;
}
int main(){
	ios::sync_with_stdio(false);
	int n, a;
	cin >> n;
	for(int i = 1; i <= n; i++){
		cin >> a;
		sum[i] = sum[i-1] + (ll)a;
	}
	bulid(1, 1, n);
	int res = 0;
	for(int i = 0; i <= n; i++){
		int c = query(1, 1, n, sum[i]);
		if(c) res = max (res, c - i);
	}
	cout << res << endl;
	return 0;
}