好消息，坏消息（洛谷 P2629）

好消息，坏消息

题目描述

uim在公司里面当秘书，现在有n条消息要告知老板。每条消息有一个好坏度，这会影响老板的心情。告知完一条消息后，老板的心情等于之前老板的心情加上这条消息的好坏度。最开始老板的心情是0，一旦老板心情到了0以下就会勃然大怒，炒了uim的鱿鱼。

uim为了不被炒，知道了了这些消息（已经按时间的发生顺序进行了排列）的好坏度，希望研究如何不让老板发怒。

uim必须按照时间的发生顺序逐条将消息告知给老板。不过uim可以使用一种叫“倒叙”的手法，例如有n条消息，小a可以从k,k+1,k+2…n,1,2…k-1这种顺序通报。

他希望知道，有多少个k，从k开始通报到n然后从1通报到k-1可以让老板不发怒。

输入格式

第一行一个整数n(1 <= n <= 10^6)，表示有n个消息。

第二行n个整数，按时间顺序给出第i条消息的好坏度Ai(-1000 <= Ai <= 1000)

输出格式

一行一个整数，表示可行的方案个数。

输入 #1

4
-3 5 1 2

输出 #1

2

样例解释

[5 1 2 -3]或[1 2 -3 5]

单调队列加前缀和的题目；先拆环成链，求得1到2n-1的前缀和sum；我们要计算1到n，2到n+1，3到n+2,等等的和，显然要求1到n的和在1到n的过程中不能为小于0，我们不可能依次遍历1到n来判断和是否小于0，我们只需算1到n最小的前缀和是否小于0就行，那么就要借助单调队列来维护一下了，先把1到n的前缀和依次入队，再依次枚举n+1到2n-1的前缀和；
代码：

#include<bits/stdc++.h>
#define LL long long
using namespace std;
int a[2000100];
LL sum[2000100];
struct Node{
	int post;
	LL date;
}sta[2000100];
int s;
int main(){
	int n;
	scanf("%d",&n);
	for(int i=1;i<=n;i++){
		scanf("%d",&a[i]);
		a[i+n]=a[i];
	} 
	for(int i=1;i<=2*n-1;i++){
		sum[i]=sum[i-1]+(LL)a[i];
	}
	int head=1,ans=0;
	for(int i=1;i<=n;i++){
		while(head<=ans&&sta[ans].date>=sum[i]){
			ans--;//出队 
		}
		sta[++ans].date=sum[i];//入队 
		sta[ans].post=i;
	}
	if(sta[head].date>=0){//判断队首是否大于0 
		s++;
	}
	for(int i=n+1;i<=2*n-1;i++){
		while(head<=ans&&sta[ans].date>=sum[i]){
			ans--;
		}
		sta[++ans].date=sum[i];
		sta[ans].post=i;
		while(head<=ans&&sta[head].post<i-n+1){//判断队首元素的位置是否符合条件 
			head++;
		}
		if(sta[head].date>=sum[i-n]){//队首是前缀和，不能直接判断是否大于0 
			s++;
		}
	}
	printf("%d\n",s);
	return 0;
}