求出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;
}