描述
小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
3
#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;
}