Week13—环中和最大的连续序列

问题描述

TT 猫咖的生意越来越红火，人越来越多，也越来越拥挤。

为了解决这个问题，TT 决定扩大营业规模，但猫从哪里来呢？

TT 第一时间想到了神秘人，想要再次通过完成任务的方式获得猫咪。

而这一次，神秘人决定加大难度。

给定一个环，A[1], A[2], A[3], … , A[n]，其中 A[1] 的左边是 A[n]。要求从环上找出一段长度不超过 K 的连续序列，使其和最大。

这一次，TT 陷入了沉思，他需要你们的帮助。
【输入】
第一行一个整数 T，表示数据组数，不超过 100。

每组数据第一行给定两个整数 N K。（1 ≤ N ≤ 100000, 1 ≤ K ≤ N）

接下来一行，给出 N 个整数。（-1000 ≤ A[i] ≤ 1000）。
【输出】
对于每一组数据，输出满足条件的最大连续和以及起始位置和终止位置。

如果有多个结果，输出起始位置最小的，如果还是有多组结果，输出长度最短的。
【样例输入】

4
6 3
6 -1 2 -6 5 -5
6 4
6 -1 2 -6 5 -5
6 3
-1 2 -6 5 -5 6
6 6
-1 -1 -1 -1 -1 -1

【样例输出】

7 1 3
7 1 3
7 6 2
-1 1 1

问题分析

1、这道题要将环路化为链路，除了最后一个元素以外的其他元素在最后一个元素后面再写一遍,保证包括所有的环序列
2、采用贪心算法，计算出现序列中2*n-1个元素的前缀和,那么连续子序列[l,r]=sum[r]-sum[l-1],在每一个r上,我只需要求出最小的范围内的l-1就可以
3、单调队列是计算每一个窗口范围内的最大(小)值得办法,首先把大于窗口范围得值踢出去,之后从另一个边把大于当前值得值请出去,再加入自己,这时在窗口内的第一个值就是窗口内最小值一直维护就一直得到当前窗口最小值,使用双向队列deque来维护单调队列
4、如果有多个结果，输出起始位置最小的，如果还是有多组结果，输出长度最短的

代码

#include<iostream>
#include<algorithm>
#include<cstring>
#include<deque>

using namespace std;
const int inf=1E9;
const int nmax=100010;
const int mmax=200020;
int A[mmax],sum[mmax];
int main(){
	int T,n,k;
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d%d",&n,&k);
		memset(A,0,sizeof(A));
		memset(sum,0,sizeof(sum));
		for(int i=1;i<=n;i++)
			scanf("%d",A+i);
		for(int i=1,temp=n+1;i<n;i++,temp++)
			A[temp]=A[i];
		int m=2*n-1;//计算前缀和
		for(int i=1;i<=m;i++){
			sum[i]=sum[i-1]+A[i];
		}
		long long ma=sum[1];
		int l=0,r=1;
		deque<int> deq;
		deq.push_front(0);
		for(int i=1;i<=m;i++)
		{
			while(!deq.empty()&&i-deq.back()>k) 
				deq.pop_back();
			if(sum[i]-sum[deq.back()]>ma)
			{
				ma=sum[i]-sum[deq.back()];
				l=deq.back();
				r=i;
			}
			else if(sum[i]-sum[deq.back()]==ma)
			{
				if(l==deq.back()) 
				{	
					if(r-l>i-deq.back()) 
					l=deq.back();
					r=i;
				}
				else if(l>deq.back())
				{
					l=deq.back();
					r=i;
				}
			}
			while(!deq.empty()&&sum[i]<sum[deq.front()])
				deq.pop_front();
			deq.push_front(i);
		}
		l++;
		if(l>n) l=l%n;
		if(r>n) r=r%n;
		printf("%lld %d %d\n",ma,l,r);
	}
}