Description
小白在玩一个游戏。桌子上有n张多米诺骨牌排成一列。它有k次机会,每次可以选一个还没有倒的骨牌,向左或者向右推倒。每个骨牌倒下的时候,若碰到了未倒下的骨牌,可以把它推倒。小白现在可以随意设置骨牌的高度,但是骨牌高度为整数,且至少为1,并且小白希望在能够推倒所有骨牌的前提下,使所有骨牌高度的和最小
Input
第一行输入一个整数T(1≤T≤10),每组数据有两行
第一行有两个整数n和k,分别表示骨牌张数和机会次数(2≤k,n≤100000)
第二行有n-1个整数,分别表示相邻骨牌的距离d(1≤d≤100000)
Output
对于每组数据,输出一行,最小的高度和
Sample Input
1
4 2
2 3 4
Sample Output
9
Solution
简单贪心,只考虑往右推,k次机会就把n张骨牌分成了k份,每份的骨牌高度和就是间距和+数量,而有k-1个间距不会被累加到答案中,所以选取前k-1大的间距不加到答案中即可
Code
#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
#define maxn 111111
int T,n,k,d[maxn];
int main()
{
scanf("%d",&T);
while(T--)
{
scanf("%d%d",&n,&k);
for(int i=0;i<n-1;i++)scanf("%d",&d[i]);
sort(d,d+n-1);
ll ans=n;
for(int i=0;i<n-1;i++)
{
if(n-1-i<k)continue;
ans+=d[i];
}
printf("%I64d\n",ans);
}
return 0;
}