题目链接:
http://acm.hdu.edu.cn/showproblem.php?pid=4193
题目大意:
给n个数,a0,a1,...an,求ai,ai+1,...an,a1,a2,...ai-1这样的排列种数,使得所有的前k(1<=k<=n)个的和都大于等于0;
解题思路:
求前缀和,加倍序列。
要满足前k个和都>=0,只需最小值>=0,所以用单调队列维护一个最小的前缀和sum[i],(i>=j-n+1),这样就保证了sum[j]-sum[i]最大,所以区间【j-n+1,i]最小。
代码:
#include<iostream>
#include<cmath>
#include<cstdio>
#include<cstdlib>
#include<string>
#include<cstring>
#include<algorithm>
#include<vector>
#include<map>
#include<set>
#include<stack>
#include<list>
#include<queue>
#include<ctime>
#define eps 1e-6
#define INF 0x3fffffff
#define PI acos(-1.0)
#define ll __int64
#define lson l,m,(rt<<1)
#define rson m+1,r,(rt<<1)|1
#pragma comment(linker, "/STACK:1024000000,1024000000")
using namespace std;
#define Maxn 1100000
int sum[Maxn<<1];
int q[Maxn];
int main()
{
//freopen("in.txt","r",stdin);
//freopen("out.txt","w",stdout);
int n;
while(scanf("%d",&n)&&n)
{
sum[0]=0;
for(int i=1;i<=n;i++) //求前缀和,并加倍序列
{
scanf("%d",&sum[i]);
sum[i+n]=sum[i];
sum[i]+=sum[i-1];
}
for(int i=n+1;i<2*n;i++)
sum[i]+=sum[i-1];
int head=0,tail=-1,ans=0; //开始队列为空
//q[0]=0;
for(int i=1;i<2*n;i++)
{
while(head<=tail&&sum[i]<=sum[q[tail]])
tail--;
q[++tail]=i;//以i结尾
while(q[head]<i-n+1)
head++;
if(i>=n) //看是否存在
{
int t=sum[i]-sum[q[head]]; //后面的最大
if(sum[i]-sum[i-n]-t>=0) //前面一块就最小,最小的大于等于0时,
ans++; //肯定满足题目要求
}
}
printf("%d\n",ans);
}
return 0;
}