トップページに戻る    次の競技プログラミングの問題へ    前の競技プログラミングの問題へ

CODE FESTIVAL 2015予選B C 旅館

■■■問題■■■

高橋君は部屋がN室ある旅館を経営しています。
今日はM組の予約が入っていますが、
全ての予約を適切に部屋に割り振ることができるかを確認していませんでした。

各予約について、その組の人数以上の定員の部屋を割り振る必要があります。
各予約について必ず1つの部屋が割り当てられる必要があり、
1つの予約について複数の部屋を割り当てたり、
複数の予約を1つの部屋に割り当てたりすることはできません。

全ての予約に部屋を割り振ることができるならばYESを、
割り振ることができないならばNOを出力してください。

■■■入力■■■

N M
A1 A2 ・・・ AN
B1 B2 ・・・ BM

●1行目には、2つの整数 N(1 <= N <= 10万), M(1 <= M <= 10万) が空白区切りで与えられる。
●2行目には、N個の整数が空白区切りで与えられる。
  このうち i(1 <= i <= N) 番目には、i番目の部屋の定員を表す整数 Ai(1 <= Ai <= 10万) が与えられる。
●3行目には、M個の整数が空白区切りで与えられる。
  このうち i(1 <= i <= M) 番目には、i番目の予約の人数を表す整数 Bi(1 <= Bi <= 10万) が与えられる。

■■■出力■■■

全ての予約に部屋を割り振ることができるならばYESを、
割り振ることができないならばNOを1行に出力せよ。
出力の末尾に改行を入れること。


C#のソース

using System;
using System.Collections.Generic;
using System.Linq;

class Program
{
    static string InputPattern = "Input1";

    static List<string> GetInputList()
    {
        var WillReturn = new List<string>();

        if (InputPattern == "Input1") {
            WillReturn.Add("3 2");
            WillReturn.Add("2 2 3");
            WillReturn.Add("3 1");
            //YES
        }
        else if (InputPattern == "Input2") {
            WillReturn.Add("3 2");
            WillReturn.Add("2 2 3");
            WillReturn.Add("3 3");
            //NO
        }
        else if (InputPattern == "Input3") {
            WillReturn.Add("3 4");
            WillReturn.Add("10 10 10");
            WillReturn.Add("1 1 1 1");
            //NO
        }
        else if (InputPattern == "Input4") {
            WillReturn.Add("10 10");
            WillReturn.Add("10 9 8 7 6 5 4 3 2 1");
            WillReturn.Add("10 9 8 7 6 5 4 3 2 1");
            //YES
        }
        else {
            string wkStr;
            while ((wkStr = Console.ReadLine()) != null) WillReturn.Add(wkStr);
        }
        return WillReturn;
    }

    static void Main()
    {
        List<string> InputList = GetInputList();
        int[] AArr = InputList[1].Split(' ').Select(X => int.Parse(X)).ToArray();
        int[] BArr = InputList[2].Split(' ').Select(X => int.Parse(X)).ToArray();

        Array.Sort(AArr);
        Array.Sort(BArr);

        int IndA = 0;
        int UB_A = AArr.GetUpperBound(0);

        for (int IndB = 0; IndB <= BArr.GetUpperBound(0); IndB++) {
            //部屋を割り当てる
            while (true) {
                if (IndA > UB_A) {
                    Console.WriteLine("NO");
                    return;
                }

                if (AArr[IndA] >= BArr[IndB]) {
                    IndA++;
                    break;
                }
                IndA++;
            }
        }
        Console.WriteLine("YES");
    }
}


解説

貪欲法で予約に対して、部屋を割り当ててます。