洛谷[P5638 【CSGRound2】光骓者的荣耀] {前缀和与差分} 奋斗的珂珂~

洛谷[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;
}