题目描述
小 K 打下的江山一共有n个城市,城市i和城市i+1有一条双向高速公路连接,走这条路要耗费时间ai。
小 K 为了关心人民生活,决定定期进行走访。他每一次会从1号城市到n号城市并在经过的城市进行访问。其中终点必须为城市n。
不仅如此,他还有一个传送器,传送半径为k,也就是可以传送到i−k和i+k。如果目标城市编号小于1则为1,大于n则为n。
但是他的传送器电量不足,只能传送一次,况且由于一些原因,他想尽量快的完成访问,于是就想问交通部部长您最快的时间是多少。
注意:他可以不访问所有的城市,使用传送器不耗费时间。
输入格式
两行,第一行n,k。
第二行n−1个整数,第i个表示ai。
输出格式
一个整数,表示答案。
输入输出样例
输入 #1
4 0
1 2 3
输出 #1
6
输入 #2
4 1
1 2 3
输出 #2
3
说明/提示
样例解释 1:
不使用传送器直接走,答案为6,可以证明这个是最小值。
样例解释 2:
在3处使用,传送到4,答案为3,可以证明这个是最小值。
思路:
记录一下前缀和,扫一遍找到节省时间最大值,答案即是1到n之间步行用时减去传送节省时间。
代码:
#include <iostream>
#include <stdio.h>
using namespace std;
const int maxn = 1e6+5;
long long a[maxn], sum[maxn];
int main()
{
int n,k;
a[0] = 0; sum[1] = 0;
cin >> n >> k;
for(int i = 2; i <= n; i++)
{
cin >> a[i];
sum[i] = sum[i-1] + a[i];
}
long long Max = 0;
for(int i = k; i <= n; i++)
{
if(sum[i] - sum[i-k] > Max)
{
Max = sum[i] - sum[i-k];
}
}
cout << sum[n] - Max << endl;
return 0;
}