洛谷[P5638 【CSGRound2】光骓者的荣耀] {前缀和与差分}
题目背景
小 K 又在做白日梦了。他进入到他的幻想中,发现他打下了一片江山。
题目描述
小 K 打下的江山一共有n个城市,城市i和城市i+1有一条双向高速公路连接,走这条路要耗费时间ai 。
小 K 为了关心人民生活,决定定期进行走访。他每一次会从1号城市到n号城市并在经过的城市进行访问。其中终点必须为城市n。
不仅如此,他还有一个传送器,传送半径为kk,也就是可以传送到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:
样例 1,2 的图示均为以下图片:
不使用传送器直接走,答案为6,可以证明这个是最小值。
样例解释 2:
在3处使用,传送到4,答案为3,可以证明这个是最小值。
数据范围:
对于所有数据,ai >0
解题思路
显而易见,要想求解最快时间就是要让传送带经过的长度尽可能的长。此时也就是要求固定跨度内长度的大小,使用前缀和可以提高算法效率。
注意
在AC之后看别人写的题解时,发现有的同学忽略了这样一个问题。题目中有这样一句话“如果目标城市编号小于1则为1,大于n则为n。”也就是说如果从此处往1传送,即使i-1<k,但是却是最大值,那么也会求出最优解。也就是说传送带不一定要传送k个城市,小于等于k也是可以的。
eg:
输入:
4 2
10 2 2
输出:
10
完整代码
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int n,k;
ll a[1000010],b[1000010],c[1000010];
//a[i]来记录输入的数据,b[i]来记录前缀和,c[i]来记录区间的距离值
int main()
{
scanf("%d %d",&n,&k);
for(int i=1;i<=n-1;i++)
{
scanf("%lld",&a[i]);//a[i]代表从i城市到i+1城市的距离
b[i]=b[i-1]+a[i];
}
if(k==0) //如果输入的是0,代表无法跳转,直接结束输出总的距离
{
printf("%lld",b[n-1]);
return 0;
}
ll ans=0;//来记录最大值
int p=0;//来记录最大值的下标
for(int i=1;i<=n-1;i++)
{
if(i>=k&&i<=n-1)//i从k开始进行,保证下标计算时不为负
c[i]=b[i]-b[i-k];
else c[i]=b[i];//如果i<k也要进行记录
//考虑如果从此处往1传送,即使i-1<k,但是却是最大值,那么也会求出最优解 。eg:第一行输入 4 2 第二行输入 10 2 2
if(ans<c[i])
{
p=i;
ans=c[i];
}
}
if(p<k) printf("%lld",b[n-1]-a[p]);//用总的距离减去跳跃过的那段距离就是结果
else printf("%lld",b[n-1]-c[p]);
return 0;
}