CodeQUEEN 2026 決勝 C 王冠づくり

問題へのリンク

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("5 4");
            WillReturn.Add("1 10");
            WillReturn.Add("2 8");
            WillReturn.Add("1 7");
            WillReturn.Add("3 5");
            WillReturn.Add("2 1");
            //30
        }
        else if (InputPattern == "Input2") {
            WillReturn.Add("5 5");
            WillReturn.Add("1 10");
            WillReturn.Add("1 9");
            WillReturn.Add("1 8");
            WillReturn.Add("2 7");
            WillReturn.Add("2 6");
            //-1
        }
        else if (InputPattern == "Input3") {
            WillReturn.Add("12 8");
            WillReturn.Add("1 10");
            WillReturn.Add("1 7");
            WillReturn.Add("1 6");
            WillReturn.Add("1 1");
            WillReturn.Add("2 9");
            WillReturn.Add("2 8");
            WillReturn.Add("2 3");
            WillReturn.Add("3 11");
            WillReturn.Add("3 5");
            WillReturn.Add("3 4");
            WillReturn.Add("4 2");
            WillReturn.Add("5 12");
            //68
        }
        else {
            string wkStr;
            while ((wkStr = Console.ReadLine()) != null) WillReturn.Add(wkStr);
        }
        return WillReturn;
    }

    static long[] GetSplitArr(string pStr)
    {
        return (pStr == "" ? new string[0] : pStr.Split(' ')).Select(pX => long.Parse(pX)).ToArray();
    }

    struct CVInfoDef
    {
        internal long Color;
        internal long Value;
    }
    static List<CVInfoDef> mCVInfoList = new List<CVInfoDef>();

    static void Main()
    {
        List<string> InputList = GetInputList();
        long[] wkArr = GetSplitArr(InputList[0]);
        long K = wkArr[1];

        Action<string> SplitAct = (pStr) => wkArr = GetSplitArr(pStr);
        foreach (string EachStr in InputList.Skip(1)) {
            SplitAct(EachStr);
            CVInfoDef WillAdd;
            WillAdd.Color = wkArr[0];
            WillAdd.Value = wkArr[1];
            mCVInfoList.Add(WillAdd);
        }

        // 同じ色での取得数上限
        long CntLimit = K / 2;

        // 取得数[色]なDict
        var CntDict = new Dictionary<long, long>();
        foreach (long EachColor in mCVInfoList.Select(pX => pX.Color)) {
            CntDict[EachColor] = 0;
        }

        var AnswerList = new List<long>();
        foreach (CVInfoDef EachCVInfo in mCVInfoList.OrderByDescending(pX => pX.Value)) {
            long CurrColor = EachCVInfo.Color;
            if (CntDict[CurrColor] == CntLimit) continue;
            CntDict[CurrColor]++;
            AnswerList.Add(EachCVInfo.Value);

            if (AnswerList.Count == K) break;
        }

        if (AnswerList.Count == K) {
            Console.WriteLine(AnswerList.Sum());
        }
        else {
            Console.WriteLine(-1);
        }
    }
}

解説

配置は後回しで個数だけを考えます。
Kの偶奇で場合分けします。

Kが6の場合
□□□□□□
同じ色が4色あったら詰みます。

Kが5の場合
□□□□□
同じ色が3色あったら詰みます。

以上により同じ色は、
(K / 2) 以下であれば良いと分かります。

あとは、個数制限をチェックしつつ、
価値の降順に選択すれば良いです。