CF-Round #637-div2-B题

CF-Round #637-div2-B题

B. Nastya and Door

传送门

这道题是贪心+前缀和~
最近效率低，没写啥题。。主要搞学校的一些环境搞得头疼。。这个星期白过了。。qaq

题目大意：有n座山，山的高度给你了。你现在要把一个门打开，你没有别的办法，只有把门仍在山上，用山峰把门撞开，已知门的长度为k，山峰是这样定义的，当一座山的高度比前一座山的高度高并且也比后一座山的高度高时，这座山就是山峰。山峰一个把一个物体分为两半，现在让你选择把长度为k的门扔在这些山的哪些地方，可以使得门的碎片数最多。输出碎片数的大小和把门扔在山上的最小左边界，下标从1开始，定义如果把门扔在区间[l, l + k - 1]的区间中，l和l + k - 1不能算作山峰。

我们把门扔在长度为k的区间中山峰数目最多的区间内即可，那么分成的碎片数就是区间的山峰数+1（比如5个山峰可以把门分成6个碎片），所以我们用前缀和来表示即可。先找到山峰，然后更新前缀和。
最后我们求长度为k的区间的最大山峰数，维护一下最大值和更新最小左边界。特别注意区间的边界不是山峰。
代码部分是从下标为0开始的，这样的话需要注意一下可能没有山峰的情况，所以pos初始化是1

代码部分：

#include <bits/stdc++.h>
#define mst(a, n) memset(a, n, sizeof(a)) 
using namespace std;
const int N = 2e5 + 10;

int n, k;
int a[N];
int c[N];
 
int main()
{
	int t;
	cin >> t;
	while (t--)
	{
		mst(c, 0);
		int cnt = 0;
		int maxx = 0;
		int pos = 1;
		scanf ("%d%d", &n, &k);
		for (int i = 0; i < n; i++)
		{
			scanf ("%d", &a[i]);
		}
		for (int i = 1; i < n - 1; i++)
		{
			if (a[i] > a[i - 1] && a[i] > a[i + 1])
			{
				c[i] = 1;
			}
			c[i] += c[i - 1];
		}
		for (int i = 0; i + k - 1 < n; i++)
		{
			if (maxx < c[i + k - 2] - c[i])
			{
				pos = i + 1;
			}
			maxx = max(maxx, c[i + k - 2] - c[i]);
		}
		cout << maxx + 1 << " " << pos << endl;
	} 
	return 0;
}