ABC358-D Souvenirs

問題へのリンク

C#のソース

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

class Program
{
    static string InputPattern = "InputX";

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

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

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

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

        var Que = new Queue<long>(AArr);

        long Answer = 0;
        foreach (long EachB in BArr) {
            while (true) {
                if (Que.Count == 0) {
                    Console.WriteLine(-1);
                    return;
                }
                if (Que.Peek() >= EachB) {
                    Answer += Que.Dequeue();
                    break;
                }
                Que.Dequeue();
            }
        }
        Console.WriteLine(Answer);
    }
}

解説

Aを昇順でキューに入れ、
Bの昇順にチェックしてます。

Queueクラスは、コンストラクタで列挙を指定することができます。