[USACO1.3] 修理牛棚 Barn Repair（贪心）

[USACO1.3] 修理牛棚 Barn Repair

题目描述

在一个月黑风高的暴风雨夜，Farmer John 的牛棚的屋顶、门被吹飞了 好在许多牛正在度假，所以牛棚没有住满。

牛棚一个紧挨着另一个被排成一行，牛就住在里面过夜。有些牛棚里有牛，有些没有。 所有的牛棚有相同的宽度。

自门遗失以后，Farmer John 必须尽快在牛棚之前竖立起新的木板。他的新木材供应商将会供应他任何他想要的长度，但是吝啬的供应商只能提供有限数目的木板。 Farmer John 想将他购买的木板总长度减到最少。

给出 $m,s,c$，表示木板最大的数目、牛棚的总数、牛的总数；以及每头牛所在牛棚的编号，请算出拦住所有有牛的牛棚所需木板的最小总长度。

输入格式

一行三个整数 $m,s,c$，意义如题目描述。
接下来 $c$ 行，每行包含一个整数，表示牛所占的牛棚的编号。

输出格式

输出一行一个整数，表示所需木板的最小总长度。

样例 #1

样例输入 #1

4 50 18
3 
4 
6 
8 
14
15 
16 
17 
21
25 
26 
27 
30 
31 
40 
41 
42 
43

样例输出 #1

25

提示

【数据范围】
对于 $100\%$ 的数据，$1\le m\le 50$，$1\le c\le s\le 200$。

USACO Training Section 1.3

思路

分析题目意思：

• 所有的牛棚有相同的宽度。
• 而且牛棚是连在一起的，要用木板把它们遮住。
• 木板最大数目是m，长度是所有的木板的长度之和。
• 个人感觉这么贪心：对于每个分界点，如果二者差距大于已有木板长度，则新开一个木板，所以我们就对差距排序。（而且我们还知道，木板越多越好）。
• 所以，我的代码的总思路就是先对读入的的牛的编号进行去重（因为有可能重复），然后用数组装入差值，接着对差值进行排序，最后对差值较大的进行标记，最后就累加即可。（这样子算的还是很快的）

代码

#include<iostream>
#include<algorithm>
#include<vector>

#define x first
#define y second

using namespace std;

const int N = 3010;

typedef pair<int,int>PII;

vector<int>a;
PII w[N];//差距数组
bool st[N];
int m,s,c;

int main(){
    cin>>m>>s>>c;
    
    
    for(int i=1;i<=c;i++){
        int x;
        cin>>x;
        a.push_back(x);
    }
    
    sort(a.begin(),a.end());
    a.erase(unique(a.begin(),a.end()),a.end());
    
    c=a.size();
    
    int j=1;
    for(int i=0;i<c-1;i++,j++){
        w[j].x=a[i+1]-a[i];
        w[j].y=i;
    }
    
    
    sort(w+1,w+1+j,greater<PII>());
    
    // cout<<w[1].y;
    
    int res=0;
    
    
    for(int i=1;i<=m-1;i++){
        // cout<<w[i].x<<" ";
        st[w[i].y]=true;
    }
    st[c-1]=true;
    int last=a[0];
    for(int i=0;i<c;i++){
        if(st[i]){
            // cout<<res<<" ";
            // cout<<a[i]<<" "<<last<<endl;
            res+=(a[i]-last+1);
            last=a[i+1];
        }
    }
    
    cout<<res;
    
    return 0;
}