链接:https://ac.nowcoder.com/acm/contest/949/D
来源:牛客网
题目描述
水果店里有 n个水果排成一列。店长要求顾客只能买一段连续的水果。
小阳对每个水果都有一个喜爱程度 ai,最终的满意度为他买到的水果的喜欢程度之和。
如果和为正(不管是正多少,只要大于 0 即可),他就满意了。
小阳想知道在他满意的条件下最多能买多少个水果。
你能帮帮他吗?
输入描述:
第一行输入一个正整数 n,表示水果总数。
第二行输入 n 个整数 ai,表示小阳对每个水果的喜爱程度。
输出描述:
一行一个整数表示结果。(如果 1 个水果都买不了,请输出 0)
示例1
输入
5
0 0 -7 -6 1
输出
1
备注:
1 ≤ n ≤ 2×106 , |ai| ≤ 103
分析:
先预处理求前缀和sum[i];
那么一段连续序列 [ L, R ] 符合就是指:sum[R] - sum[L-1] > 0,即 sum[R] > sum[L-1]
其序列长度为 R - L + 1;
那么对于以 i 结尾的最长序列,就是要找到第一个(即最远的一个) sum[j] < sum[i] 的位置 j,则可求得长度为 i - j;
但是这样之间依次查询会令时间复杂度达到O(N2);
这里就要用到单调队列,需要维护一个sum值递减的单调队列,这样每次只需要 二分upper_bound() 就可以找到位置 j;
维护方法:
对于sum[i],若小于队尾元素,则入队,否则不入队。(若队列为空,直接入队)
①因为是按顺序依次入列的,所以二分找到的第一个小于sum[i]的元素位置必定是最远的一个。
②至于为什么sum[i]大于等于队尾元素不入队,是因为队尾元素既比sum[i]更小,又比其更先入列(即位置更远,得到的长度更长)。
以下代码:
#include<iostream