单调队列——HDU 3415 Max Sum of Max-K-sub-sequence

最新推荐文章于 2020-09-22 16:18:37 发布

庞老板

最新推荐文章于 2020-09-22 16:18:37 发布

阅读量889

点赞数

分类专栏：数据结构文章标签：单调队列 HDU 3415

本文链接：https://blog.csdn.net/code_pang/article/details/14528501

版权

数据结构专栏收录该内容

10 篇文章 0 订阅

订阅专栏

Max Sum of Max-K-sub-sequence

Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 5216 Accepted Submission(s): 1890

Problem Description

Given a circle sequence A[1],A[2],A[3]......A[n]. Circle sequence means the left neighbour of A[1] is A[n] , and the right neighbour of A[n] is A[1].
Now your job is to calculate the max sum of a Max-K-sub-sequence. Max-K-sub-sequence means a continuous non-empty sub-sequence which length not exceed K.

Input

The first line of the input contains an integer T(1<=T<=100) which means the number of test cases.
Then T lines follow, each line starts with two integers N , K(1<=N<=100000 , 1<=K<=N), then N integers followed(all the integers are between -1000 and 1000).

Output

For each test case, you should output a line contains three integers, the Max Sum in the sequence, the start position of the sub-sequence, the end position of the sub-sequence. If there are more than one result, output the minimum start position, if still more than one , output the minimum length of them.

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

题目与单调队列—— HDU 4193 Non-negative Partial Sums类似，不在赘述，直接给出代码：

#include <stdio.h>

#define MAX_NUM	100000
#define NEGATIVE_INF -0x7ffffff

int sum[MAX_NUM*2+5];	//存储循环序列的前段和
int n;					//序列长度
int k;					//子序列的最大长度

int mq[MAX_NUM+5];		//单调不增队列，队列元素为a[]的下标
int front;				//队首指针
int rear;				//队尾指针，指向队尾的下一个位置

inline void InitMonotonicQueue(void)
{
	front = rear = 0;
}

inline bool IsEmpty(void)
{
	return front == rear;
}

void Push(int i)
{
	while (!IsEmpty() && sum[mq[rear-1]] < sum[i])
	{
		rear--;
	}
	mq[rear++] = i;
}

inline int Front(void)
{
	return mq[front];
}

inline void Pop(void)
{
	front++;
}

int main(void)
{
	int ncases;
	scanf("%d", &ncases);
	
	while (ncases-- != 0)
	{
		scanf("%d%d", &n, &k);
		int i;
		for (i = 1; i <= n; i++)
		{
			scanf("%d", &sum[i]);
		}

		for (i = n+1; i < n+k; i++)
		{
			sum[i] = sum[i-n];
		}

		for (i = 2; i < n+k; i++)
		{
			sum[i] += sum[i-1];
		}

		InitMonotonicQueue();
		for (i = 1; i <= k; i++)
		{
			Push(i);
		}
		int maxSum = sum[Front()];
		int startPos = 1, endPos = Front();

		for (i = k+1; i < n+k; i++)
		{
			if (!IsEmpty() && Front() + k <= i)
			{
				Pop();
			}
			Push(i);
			int p = Front();
			int s = sum[p] - sum[i-k];
			if (s > maxSum)
			{
				maxSum = s;
				startPos = i - k + 1;
				endPos = p;
			}
		}
		if (endPos > n)
		{
			endPos = endPos % n;
		}
		printf("%d %d %d\n", maxSum, startPos, endPos);
	}
	return 0;
}