ACM 拆除土墙

最新推荐文章于 2024-02-25 18:23:11 发布

原创最新推荐文章于 2024-02-25 18:23:11 发布 · 260 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#ACM

ACM 专栏收录该内容

8 篇文章

订阅专栏

描述小A家的大门外有一堵土墙，这堵土墙是由n个高度可能不相同的长方体从左至右砌成的。例如，当n为7，h=[1,2,6,1,1,7,1]时，这堵墙的图形如下：

某日，小A从外面买了一台宽度为k的电视，他想要将这个电视搬进家里，所以，他必须要从这堵土墙中拆除k个连续的长方体，以至于能将宽度为k的电视搬进家里。但是，小A又想这k个连续的长方体高度之和最小，以至于拆除的土墙最不费力。
小A想问问你，从哪块长方体开始拆，以至于这连续的k块长方体高度之和最小。

输入有多组输入数据，每组第一行包含两个正整数n，k（1<=n<=1.5*10^5，1<=k<=n），n代表这堵土墙由n个高度可能不相同的长方体组成的，k代表电视机的宽度。第二行有n个整数，h1,h2,..,hn(1<=hi<=100)，代表从左至右这n个长方体的高度。

输出输出一个整数j，表示从j块长方体开始拆，j,j+1,...,j+k-1这k块长方体的高度和最小。如果j有多个位置，输出最左边的位置。

样例输入1

7 3
1 2 6 1 1 7 1

样例输出1

#include<cstdio>
#include<cstring>//memset函数头文件
#define MAX 1e9//1e9表示10^9，这题最大范围是150000*100，所以10^9来判断绝对是够用的
int main()
{
	//freopen("data.in","r",stdin);
	int n, k;
	int h[150010];
	while (scanf("%d%d", &n, &k) != EOF)
	{
		int sum[150010];//前缀和函数
		memset(sum, 0, sizeof(sum));
		for (int i = 1; i <= n; i++)
		{
			scanf("%d", &h[i]);
			sum[i] = sum[i - 1] + h[i];//这里有个小技巧，sum数组存下前缀和，在后续降低时间复杂度很有用
		}
		int minn = MAX, pos = 0;//pos存位置，minn存最小的高度和
		for (int i = k; i <= n; i++)
			if (sum[i] - sum[i - k]<minn)
			{
				minn = sum[i] - sum[i - k];
				pos = i - k + 1;
			}
		printf("%d\n", pos);
	}
	return 0;
}