题解 洛谷P1209 【[USACO1.3]修理牛棚 Barn Repair】

题面：

题目描述

在一个夜黑风高,下着暴风雨的夜晚,farmer John的牛棚的屋顶、门被吹飞了。 好在许多牛正在度假，所以牛棚没有住满。 牛棚一个紧挨着另一个被排成一行，牛就住在里面过夜。 有些牛棚里有牛，有些没有。 所有的牛棚有相同的宽度。 自门遗失以后,farmer John必须尽快在牛棚之前竖立起新的木板。 他的新木材供应商将会供应他任何他想要的长度,但是吝啬的供应商只能提供有限数目的木板。 farmer John想将他购买的木板总长度减到最少。

给出:可能买到的木板最大的数目M(1<= M<=50);牛棚的总数S(1<= S<=200); 牛棚里牛的总数C(1 <= C <=S);和牛所在的牛棚的编号stall_number(1 <= stall_number <= S),计算拦住所有有牛的牛棚所需木板的最小总长度。 输出所需木板的最小总长度作为答案。

输入格式

第 1 行: 木板最大的数目M ,牛棚的总数S 和 牛的总数C(用空格分开)

第 2 到 C+1行: 每行包含一个整数，表示牛所占的牛棚的编号。

输出格式

单独的一行包含一个整数表示所需木板的最小总长度。

---

 

可以先将这道题反过来想

可以理解为有一块木板将整个盖住，每两个有牛的棚子中间有空隙，可以断开m-1次，如图例

但还需要特判m>c的情况，因为需要的木板比消耗的木板多，会导致第6,7个点WA或RE

上代码

#include<algorithm>
#include<iostream>
#include<iomanip>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<queue>
using namespace std;
int m,s,c,a[205],b[205],ans=0;
bool cmp(int x,int y){return x>y;}//从大到小排 
int read()
{   int x=0; char ch=getchar();
    while(ch<'0'||ch>'9')ch=getchar();
    while(ch>='0'&&ch<='9') { x=x*10+ch-'0'; ch=getchar(); }
    return x;
}//快读 
int main()
{
    m=read();s=read();c=read();
    if(m>c){printf("%d",c);return 0;}//特判需要的木板比消耗的木板多的情况 
    for(int i=1;i<=c;i++)a[i]=read();
    sort(a+1,a+c+1);
    for(int i=2;i<=c;i++)b[i-1]=a[i]-a[i-1];//求出有牛的牛棚的中间的空隙 
    ans=a[c]-a[1]+1;//初始将整个盖住 
    sort(b+1,b+c,cmp);//空隙从大到小排 
    for(int i=1;i<=m-1;i++)ans=ans-b[i]+1;//减去空隙 
    printf("%d",ans);
    return 0;
}