【Week 13 作业E】TT的神秘任务3

题目描述

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

思路

这里使用前缀和的思想，首先根据输入计算前缀和数组。
因为数据是一个环，所以可以想象成数据由两份相同的数据组成，把数据再接到最后一个数据后面继续计算前缀和，这样就可以达到变环为链的效果。
对于第i个数据的前缀和，目前的最大的子串和为sum[i]-sum[j],其中j为i-k到i中最小的前缀和。这里可以采用单调栈进行优化来选取j的位置。

反思

开始时使用vector容器以及list来模拟单调栈，但都出现了超时错误，后来直接使用了数组来模拟单调栈，可以较大程度减少运行时间。

代码

#include <iostream>
#include <vector>
using namespace std;
const int size=2*1e5+10;
const int inf=1e9;
int sum[size];
int st[size];
int n,m,k;
void solve()
{
	int ans=-inf,l=0,r=0;
	//vector<int> v;
	int sl=0,sr=0;st[0]=0;
	//v.push_back(0);
	for(int i=1;i<m;i++)
	{
		while((sr-sl)>=0&&sum[st[sr]]>sum[i])
			sr--;//v.pop_back();
		st[++sr]=i;
		//v.push_back(i);
		while((sr-sl)>=0&&i-st[sl]>k)
			sl++;//v.erase(v.begin());
		if((sr-sl>0&&(sum[i]-sum[st[sl]])>ans)
			||(sr-sl==0&&sum[i]-sum[i-1]>ans))
		{
			if(sr-sl>0)
			{
				ans=sum[i]-sum[st[sl]];
				l=st[sl]+1;
				r=i;
			}
			else
			{
				ans=sum[i]-sum[i-1];
				l=r=i;
			}
		}
	}
	printf("%d %d ",ans,l);
	if(r>n)
		printf("%d\n",r-n);
	else
		printf("%d\n",r);
}
int main(int argc, char** argv) {
	int T;
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d%d",&n,&k);
		sum[0]=0;
		for(int i=1;i<=n;i++)
		{
			int temp;
			scanf("%d",&temp);
			sum[i]=sum[i-1]+temp;
		}
		m=n+1;
		for(int i=1;i<=n&&m<=n+k;m++,i++)
		{
			sum[m]=sum[m-1]+(sum[i]-sum[i-1]);			
		}
		solve();
	}
	return 0;
}