BestCoder Round #85 1002 domino

问题描述

小白在玩一个游戏。桌子上有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

思路：先按距离大小从小到大排列好，然后将前n-k个距离相加（将k-1个距离最大的放在组后面，最后用剩下的k-1一次机会一个一个推倒，此时就不用管距离了），因为骨牌高至少为1且高度必须大于等于距离+1，所以最后再加上n，即为答案.

代码如下：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int dis[100100];
int main()
{
    int n,k,t,i;
    scanf("%d",&t);
    __int64 sum;
     while(t--)
    {
        sum=0;
        scanf("%d%d",&n,&k);
        for(i=0;i<n-1;i++)
        {
            scanf("%d",&dis[i]);
        }
        sort(dis,dis+n-1);
        for(i=0;i<n-k;i++)
        sum+=dis[i];
        sum+=n;
        printf("%I64d\n",sum);
        
    }
}