您的位置:首页 > 其它

HDU-4193 Non-negative Partial Sums 单调队列

2012-10-09 00:30 369 查看
题目:http://acm.hdu.edu.cn/showproblem.php?pid=4193

题意:给定一个数列,进行绕环移动,然后问所有前缀和都大于等于零的数列有多少。

首先按照一般方法,把数列往后复制,即a1a2,,,ana1a2,,,an,然后对其求前缀和。接下来就要求移动的区间内的最小值(这里很像POJ2823),看其是否大于等于零。那么就可用到单调队列来求了,队列里保存当前区间里的单调不递减的值,那么就可以在O(1)的复杂度内知道当前区间的最小值,即队首元素大于等于零,则ans+1。

My code:

//STATUS:C++_AC_703MS_198000KB
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<algorithm>
#include<string>
#include<vector>
#include<queue>
#include<stack>
#include<set>
#define LL __int64
#define Max(x,y) ((x)>(y)?(x):(y))
#define Min(x,y) ((x)<(y)?(x):(y))
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define mem(a,b) memset(a,b,sizeof(a))
const int MAX=1000010,INF=200000000,MOD=1000000007;
const double esp=1e-6;

int num[MAX*2],sum[MAX*2],q[MAX*2];
int n;

int main()
{
//	freopen("in.txt","r",stdin);
	int i,j,front,rear,ans,max;
	while(~scanf("%d",&n) && n)
	{
		max=(n<<1);
		ans=front=rear=0;
		for(i=0;i<n;i++){
			scanf("%d",&num[i]);
			num[n+i]=num[i];
		}
		sum[0]=num[0];
		for(i=1;i<max;i++)
			sum[i]=sum[i-1]+num[i];

		for(i=0;i<n;i++){
			while(rear>front && q[rear-1]>sum[i])
				rear--;
			q[rear++]=sum[i];
		}
		if(q[0]>=0)ans++;

		for(i=0;i<n-1;i++){
			if(q[front]==sum[i])
				front++;
			while(rear>front && q[rear-1]>sum[n+i])
				rear--;
			q[rear++]=sum[n+i];
			if(q[front]-sum[i]>=0)ans++;
		}

		printf("%d\n",ans);
	}
	return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: