单调队列优化dp TT的神秘任务三

题意:

TT 猫咖的生意越来越红火,人越来越多,也越来越拥挤。
为了解决这个问题,TT 决定扩大营业规模,但猫从哪里来呢?
TT 第一时间想到了神秘人,想要再次通过完成任务的方式获得猫咪。
而这一次,神秘人决定加大难度。
给定一个环,A[1], A[2], A[3], … , A[n],其中 A[1] 的左边是 A[n]。要求从环上找出一段长度不超过 K 的连续序列,使其和最大。
这一次,TT 陷入了沉思,他需要你们的帮助。

Input
第一行一个整数 T,表示数据组数,不超过 100。
每组数据第一行给定两个整数 N K。(1 ≤ N ≤ 100000, 1 ≤ K ≤ N)
接下来一行,给出 N 个整数。(-1000 ≤ A[i] ≤ 1000)。
Output
对于每一组数据,输出满足条件的最大连续和以及起始位置和终止位置。
如果有多个结果,输出起始位置最小的,如果还是有多组结果,输出长度最短的。

Sample Input

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

Sample Output

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

思路:

首先这道题是一个环形序列,为了处理这个环形序列的,我们可以将数组大小设为2n-1,将0到n-2复制到n到2n-2处即可。

首先求出前缀和sum[i],f[i] =sum[i]-min{sum[j]}, i-k<=j<=i-1。而min{sum[j]}可以采用单调队列来求取,维护队列的长度不超过k,而sum[i] - sum[j]实际上是j+1到i,这点需要注意。

当有有多个结果,输出起始位置最小的,如果还是有多组结果,输出长度最短的。这点很好处理,只需要在判断的时候设立符号是>即可,这样只取最先碰到的,起始位置在他之后或是起始位置一样但是长度较长的一定在这个最先碰到的之后。


代码:

#define _CRT_SECURE_NO_WARNINGS
#include<iostream>
#include<deque>
#include<stdio.h>
#include<string.h>
using namespace std;

long long num[2 * 100000 + 5];
long long sum[2 * 100000 + 5];


int main()
{
	int T;
	scanf("%d", &T);
	for (int i = 0; i < T; i++)
	{
		memset(num, 0, sizeof(num));
		memset(sum, 0, sizeof(sum));

		int n, k;
		scanf("%d %d", &n, &k);
		int numLength = 2 * n - 1;

		for (int i = 1; i <= n; i++)
			scanf("%lld", &num[i]);
		for (int i = n + 1; i <= numLength; i++)
			num[i] = num[i % n];

		long long ans = -100000000;
		for (int i = 1; i <= numLength; i++)sum[i] = sum[i - 1] +  num[i];
		deque<int>q;
		q.push_back(0);
		int l, r;//代表输出队列的最左端之前和最右位置
		for (int i = 1; i <= numLength; i++)
		{
			while (!q.empty() && sum[q.back()] > sum[i])
				q.pop_back();
			q.push_back(i);
			
			while (!q.empty() && i- q.front() > k)
				q.pop_front();
			int lasa = q.front();
			long long temp;
			if (q.size() > 1 ) 
			{
				temp = sum[i] - sum[q.front()];
				if (q.front() > n)break;
				if (temp > ans)
				{
					ans = temp;
					l = q.front();
					r = i;
				}
			}
			else
			{
				temp = sum[i] - sum[i - 1];
				if (temp > ans)
				{
					ans = temp;
					l = i - 1;
					r = i;
				}
			}
		}

		if (r > n)
			printf("%lld %d %d\n", ans, l + 1, r % n);
		else
			printf("%lld %d %d\n", ans, l + 1, r % (n + 1));

	}
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值