domino
Accepts: 462
Submissions: 1498
Time Limit: 2000/1000 MS (Java/Others)
Memory Limit: 131072/131072 K (Java/Others)
问题描述
小白在玩一个游戏。桌子上有n张多米诺骨牌排成一列。它有k次机会,每次可以选一个还没有倒的骨牌,向左或者向右推倒。每个骨 牌倒下的时候,若碰到了未倒下的骨牌,可以把它推倒。小白现在可以随意设置骨牌的高度,但是骨牌高度为整数,且至少为1,并且 小白希望在能够推倒所有骨牌的前提下,使所有骨牌高度的和最小。
输入描述
第一行输入一个整数T(1≤T≤10) 每组数据有两行 第一行有两个整数n和k,分别表示骨牌张数和机会次数。(2≤k,n≤100000) 第二行有n-1个整数,分别表示相邻骨牌的距离d,1≤d≤100000
输出描述
对于每组数据,输出一行,最小的高度和
输入样例
1 4 2 2 3 4
输出样例
9
/* ***********************************************
Author : ryc
Created Time : 2016-07-31 Sunday
File Name : E:\acmcode\hdoj\5777.cpp
LANGUAGE : c++
Copyright 2016 ryc All Rights Reserved
************************************************ */
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#include<list>
#include<vector>
#include<stack>
using namespace std;
const int maxn=1000010;
long long pos[maxn];
bool vis[maxn];
struct Node{
long long x,y;
}A[maxn];
bool cmp(Node a,Node b){
return a.x>b.x;
}
int main()
{
long long n,k,t;cin>>t;
while(t--){
memset(vis,false,sizeof(vis));
scanf("%lld%lld",&n,&k);
for(int i=1;i<n;++i){
scanf("%lld",&pos[i]);
A[i].x=pos[i];A[i].y=i;
}
sort(A+1,A+n,cmp);
for(int i=1;i<=k-1&&i<n;++i){
vis[A[i].y]=true;
}
long long ans=1ll,last=0ll,cnt=1ll;
for(int i=n-1;i>=1ll;--i){
if(vis[i]&&cnt<k){
cnt++;last=1;vis[i]=false;
}
else {
last=pos[i]+1ll;
}
ans=ans+last;
}
printf("%lld\n",ans);
}
return 0;
}