洛谷P5638 【CSGRound2】光骓者的荣耀 入门

P5638 【CSGRound2】光骓者的荣耀

标签：暴力
题目背景

小 K 又在做白日梦了。他进入到他的幻想中，发现他打下了一片江山。
题目描述

小 K 打下的江山一共有nnn个城市，城市iii和城市i+1i+1i+1有一条双向高速公路连接，走这条路要耗费时间aia_iai​。

小 K 为了关心人民生活，决定定期进行走访。他每一次会从111号城市到nnn号城市并在经过的城市进行访问。其中终点必须为城市nnn。

不仅如此，他还有一个传送器，传送半径为kkk，也就是可以传送到i−ki-ki−k和i+ki+ki+k。如果目标城市编号小于111则为111，大于nnn则为nnn。

但是他的传送器电量不足，只能传送一次，况且由于一些原因，他想尽量快的完成访问，于是就想问交通部部长您最快的时间是多少。

注意：他可以不访问所有的城市，使用传送器不耗费时间
输入格式

两行，第一行n,kn,kn,k。

第二行n−1n-1n−1个整数，第iii个表示aia_iai​。
输出格式

一个整数，表示答案。
输入输出样例
输入 #1

4 0
1 2 3

输出 #1

6

输入 #2

4 1
1 2 3

输出 #2

3

说明/提示
样例解释 1：

样例 1,2 的图示均为以下图片：

不使用传送器直接走，答案为666，可以证明这个是最小值。
样例解释 2：

在333处使用，传送到444，答案为333，可以证明这个是最小值。
数据范围：

对于所有数据，ai>0a_i > 0ai​>0

#include<iostream>
#include<cstdio>
using namespace std;
#define m_ax 1000000+5
long long a[m_ax];
int main()
{
	int n,k;
	long long sum=0,max=0;
	scanf("%d%d",&n,&k);
	n=n-1;
	for(int i=0;i<n;i++)
	{
		scanf("%lld",&a[i]);
		sum=a[i]+sum;
	}
	if(n<=k)
	{
		printf("0\n");
	}
	else if(k==0){
		printf("%lld\n",sum);
	}
	else{
		long long ax=0;
		for(int i=0;i<n-k+1;i++)
		{
			if(i==0)
			{
				for(int j=i;j<k+i;j++)
				{
					ax=ax+a[j];
				}
			}
			else{
				ax=ax+a[k+i-1]-a[i-1];
			}
			
			printf("ax=%lld\n",ax);
			if(ax>max)
				max=ax;
		}
		printf("%lld\n",sum-max);
	}
	
	
	
	return 0;
}