牛客网小白月赛D

题目：
水果店里有 n个水果排成一列。店长要求顾客只能买一段连续的水果。小阳对每个水果都有一个喜爱程度 ai，最终的满意度为他买到的水果的喜欢程度之和。如果和为正（不管是正多少只要大于 0 即可），他就满意了。小阳想知道在他满意的条件下最多能买多少个水果。
你能帮帮他吗？
输入描述:
第一行输入一个正整数 n，表示水果总数。

第二行输入 n 个整数 ai，表示小阳对每个水果的喜爱程度。
输出描述:
一行一个整数表示结果。（如果 1 个水果都买不了，请输出 0）

思路1：树状数组（不知道为什么线段树超时），维护前缀和sum[i]，先离散化，不然sum[i]很大，从1开始update，找到离散化后的下标，把他代表的值更新为min（原值，传入值i），每次对i查询时只要查询小于他的sum[i]，这样sum[i]-sum[j]就能大于0，在所有小于他的j里找到最小的，所以树状数组维护的是最小值

代码

#include <iostream>
#include <algorithm>
#include <string.h>
#include <stdio.h>
using namespace std;
const int N = 4111111;
const int inf = (1<<31)-2;
int a[N],sum[N],b[N],tree[N],n;
void add(int u,int v){
    for(;u <= n; u += u & -u) tree[u] = min(tree[u],v);
}
int getmin(int u){
    int ans = inf;
    for(;u > 0; u -= u & -u) ans = min(ans,tree[u]);
    return ans;
}
int main()
{
    cin >> n;
    sum[0] = 0;
    for(int i = 0 ; i < n + 2 ; i ++)
        tree[i] = inf;
    for(int i = 1 ; i <= n ; i ++)
    {
        scanf("%d",&a[i]);
        sum[i] = sum[i-1]+a[i];
        b[i] = sum[i];
    }
    sort(sum+1,sum+1+n);
    int len = unique(sum+1,sum+1+n) - sum-1;
    int x = lower_bound(sum+1,sum+1+len,0) - sum;
    add(x,0);
    int mx = 0;
    for(int i = 1 ; i <= n ; i ++)
    {
        int pos = lower_bound(sum+1,sum+1+len,b[i])-sum;
        add(pos,i);
        mx = max(mx,i-getmin(pos-1));
    }
    cout << mx << endl;
    return 0;
}

用线段树的超时代码

#include <iostream>
#include <algorithm>
#include <string.h>
#include <stdio.h>
using namespace std;
const int N = 222222;
const int inf = (1<<31)-2;
int a[N],sum[N],b[N];
struct tree
{
    int l,r,mi;
}q[N*4];
void push_up(int n)
{
    q[n].mi = min(q[n<<1].mi,q[n<<1|1].mi);
}
void built(int n,int l,int r)
{
    q[n].l = l;
    q[n].r = r;
    if(l==r)
    {
        q[n].mi = inf;
        return;
    }
    int mid = (l+r)>>1;
    built(n<<1,l,mid);
    built(n<<1|1,mid+1,r);
    push_up(n);
}
void update(int n,int pos,int num)
{
    if(q[n].l==q[n].r)
    {
        q[n].mi = min(q[n].mi,num);
        return;
    }
    int mid = (q[n].l+q[n].r)>>1;
    if(pos<=mid)
        update(n<<1,pos,num);
    else
        update(n<<1|1,pos,num);
    push_up(n);
}
int query(int n,int l,int r)
{
    if(q[n].l>=l && q[n].r<=r)
    {
        return q[n].mi;
    }
    int mid = (q[n].l+q[n].r)>>1;
    if(r<=mid)
        return query(n<<1,l,r);
    else if(l>mid)
        return query(n<<1|1,l,r);
    else
    {
        return min(query(n<<1,l,mid),query(n<<1|1,mid+1,r));
    }
}
int main()
{
    int n;
    cin >> n;
    sum[0] = 0;
    for(int i = 1 ; i <= n ; i ++)
    {
        scanf("%d",&a[i]);
        sum[i] = sum[i-1]+a[i];
        b[i] = sum[i];
    }
    sort(sum+1,sum+1+n);
    int len = unique(sum+1,sum+1+n) - sum-1;
    int x = lower_bound(sum+1,sum+1+len,0) - sum;
    built(1,0,len+1);
    update(1,x,0);
    int mx = 0;
    for(int i = 1 ; i <= n ; i ++)
    {
        int pos = lower_bound(sum+1,sum+1+len,b[i])-sum;
        update(1,pos,i);
        mx = max(mx,i-query(1,0,pos-1));
    }
    cout << mx << endl;
    return 0;

方法2：还是线段树，这次线段树维护的是最大的sum[i]，从1开始，如果根节点大于tot（一开始是0），那就说明以1开始有大于0的某段，找出下标最大的大于tot的，一直循环到n，每次循环tot+=a[i]

代码

#include <bits/stdc++.h>
using namespace std;

const int N = 2e6 + 10;
int a[N], s[N];
int n, ans, l, r, tot;
int mx[N << 2];

void pushup(int rt) {
    mx[rt] = max(mx[rt << 1], mx[rt << 1 | 1]);
}

void build(int rt, int l, int r) {
    if(l == r) {
        mx[rt] = s[l];
        return;
    }
    int mid = l + r >> 1;
    build(rt << 1, l, mid);
    build(rt << 1 | 1, mid + 1, r);
    pushup(rt);
}

int query(int rt, int l, int r) {
    if(l == r)
        return l;
    int mid = l + r >> 1;
    if(mx[rt << 1 | 1] > tot)
        return query(rt << 1 | 1, mid + 1, r);
    else
        return query(rt << 1, l, mid);
}

int main() {
    scanf("%d", &n);
    for(int i = 1; i <= n; ++i) {
        scanf("%d", &a[i]);
        s[i] = s[i - 1] + a[i];
    }
    build(1, 1, n);
    ans = 0;
    cout << tot << endl;
    for(int i = 1; i <= n; ++i) {
        if(mx[1] <= tot) {
            tot += a[i];
            continue;
        }
        ans = max(ans, query(1, 1, n) - i + 1);
        tot += a[i];
    }
    printf("%d\n", ans);
    return 0;
}