USACO Training 1.4 Barn Repair

It was a dark and stormy night that ripped the roof and gates off the stalls that hold Farmer John's cows. Happily, many of the cows were on vacation, so the barn was not completely full.

The cows spend the night in stalls that are arranged adjacent to each other in a long line. Some stalls have cows in them; some do not. All stalls are the same width.

Farmer John must quickly erect new boards in front of the stalls, since the doors were lost. His new lumber supplier will supply him boards of any length he wishes, but the supplier can only deliver a small number of total boards. Farmer John wishes to minimize the total length of the boards he must purchase.

Given M (1 <= M <= 50), the maximum number of boards that can be purchased; S (1 <= S <= 200), the total number of stalls; C (1 <= C <= S) the number of cows in the stalls, and the C occupied stall numbers (1 <= stall_number <= S), calculate the minimum number of stalls that must be blocked in order to block all the stalls that have cows in them.

Print your answer as the total number of stalls blocked.

PROGRAM NAME: barn1

INPUT FORMAT

SAMPLE INPUT (file barn1.in)

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

OUTPUT FORMAT

A single line with one integer that represents the total number of stalls blocked.

SAMPLE OUTPUT (file barn1.out)

25

/*
ID: choiyin1
TASK: barn1
LANG: C++
*/
#include
using namespace std;
const int maxn = 200 + 5;
bool vis[maxn];
int open[maxn];
bool cmp(int x, int y)
{
    return x > y;
}
int num[maxn];
int main(){
    freopen("barn1.in", "r", stdin);
    freopen("barn1.out", "w", stdout);
    int m, s, c;
    while(scanf("%d%d%d", &m, &s, &c) == 3)
    {
        int Min = maxn, Max = 0;
        memset(vis, 0, sizeof(vis));
        for(int i = 0; i < c; i++)
        {
            int t;
            scanf("%d", &t);
            vis[t] = 1;
            Min = min(Min, t);
            Max = max(Max, t);
        } 
        int cnt = 0;
        for(int i = Min; i <= Max; i++)
        {
            if(!vis[i])
                open[cnt]++;
            else if(open[cnt]) cnt++;
        }
        int ans = 0;
        sort(open, open + cnt, cmp); 
        for(int i = 0; i < m - 1; i++)
        {
            ans = ans + open[i];
        }
        printf("%d\n", Max - Min - ans + 1);
    } 
}