hdoj5777domino

domino

 

 Accepts: 462

 

 Submissions: 1498

 Time Limit: 2000/1000 MS (Java/Others)

 

 Memory Limit: 131072/131072 K (Java/Others)

问题描述

小白在玩一个游戏。桌子上有n张多米诺骨牌排成一列。它有k次机会，每次可以选一个还没有倒的骨牌，向左或者向右推倒。每个骨
牌倒下的时候，若碰到了未倒下的骨牌，可以把它推倒。小白现在可以随意设置骨牌的高度，但是骨牌高度为整数，且至少为1，并且
小白希望在能够推倒所有骨牌的前提下，使所有骨牌高度的和最小。

输入描述

第一行输入一个整数T($1\le T\le 10$)
每组数据有两行
第一行有两个整数n和k，分别表示骨牌张数和机会次数。($2\le k,n\le 100000$)
第二行有n-1个整数，分别表示相邻骨牌的距离d，$1\le d\le 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;
}