牛客小白月赛16 D 小阳买水果 （思维题）

链接：https://ac.nowcoder.com/acm/contest/949/D
来源：牛客网

时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 262144K，其他语言524288K
64bit IO Format: %lld

题目描述

水果店里有 $\mathrm{nn个水果排成一列。店长要求顾客只能买一段连续的水果。\; 小阳对每个水果都有一个喜爱程度\; aiai，最终的满意度为他买到的水果的喜欢程度之和。\; 如果和为正（不管是正多少，只要大于\; 00\; 即可），他就满意了。}$

小阳想知道在他满意的条件下最多能买多少个水果。

$\mathrm{你能帮帮他吗？}$

输入描述:

第一行输入一个正整数 n，表示水果总数。

第二行输入 n 个整数 ${\mathrm{aiai，表示小阳对每个水果的喜爱程度。}}^{}$

输出描述:

一行一个整数表示结果。（如果 1 个水果都买不了，请输出 0）

示例1

输入

复制

5
0 0 -7 -6 1

输出

复制

1

备注:

$\mathrm{1\le n\le 2\times 106,|ai|\le 1031\le n\le 2\times 106,|ai|\le 103}$


解题思路：把前缀和从小到大排序，如果前缀和相等把编号大的放前面，然后从前往后遍历，记录遍历过程最小的编号，如果当前的编号大于最小的编号，则更新答案。注意特判n=1的情况就可以了
代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=2e6+7;
int n;
struct node{
    int pos,num;
}p[maxn];
bool cmp(node a,node b){
    if(a.num==b.num) return a.pos>b.pos;
    return a.num<b.num;
}

int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        int x;
        scanf("%d",&x);
        p[i].num=p[i-1].num+x;
        p[i].pos=i;
    }
    n++;
    int minpos=n;
    int ans=0;
    sort(p+1,p+n+1,cmp);
    for(int i=1;i<=n;i++){
        minpos=min(minpos,p[i].pos); //记录编号最小的位置
        if(minpos<p[i].pos) //当前编号大于记录的最小编号，则更新答案
        ans=max(ans,p[i].pos-minpos);
    }
    printf("%d\n",ans);
    return 0;
}