题目链接:
http://acm.hdu.edu.cn/showproblem.php?pid=3415
题意:
给定一个长度为n的环,求其中一个长度小于等于k的区间,使其和最大,输出最大和,区间的左右端点
分析:
直接想到暴力 dp[i]= sum[i] - sum[i-k+1+j] (0<=j<=k);
但是很明显暴力会超时,就想到用单点队列来维护,一个sum数组,
使其sum尽量小的在前面.处理的时候要将环展开,sum存的是前缀和。
代码如下:
#include <iostream> #include <cstring> #include <cstdio> #include <algorithm> #include <queue> using namespace std; const int maxn = 100010; int a[maxn*2],sum[maxn*2]; int main() { int t,n,k; scanf("%d",&t); while(t--){ scanf("%d%d",&n,&k); memset(sum,0,sizeof(sum)); for(int i=1;i<=n;i++){ scanf("%d",a+i); sum[i]=sum[i-1]+a[i]; a[i+n]=a[i]; } for(int i=n+1;i<=2*n;i++) sum[i]=sum[i-1]+a[i]; deque<int >Q; int ans = -1000000000,st=0,ed=0; for(int i=1;i<=n+k-1;i++){ while(!Q.empty()&&sum[i-1]<sum[Q.back()])//保持队列的单调性 Q.pop_back(); while(!Q.empty()&&Q.front()<i-k)//超过k的长度则消除队列前面的元素 Q.pop_front(); Q.push_back(i-1); if(sum[i]-sum[Q.front()]>ans){ ans=sum[i]-sum[Q.front()]; st=Q.front()+1; ed=i; } } if(ed>n) ed-=n; printf("%d %d %d\n",ans,st,ed); } return 0; }
时间: 2024-10-06 00:22:02