ARC183-A Median of Good Sequences

問題へのリンク

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("2 2");
            //1 2 2 1
        }
        else if (InputPattern == "Input2") {
            WillReturn.Add("1 5");
            //1 1 1 1 1
        }
        else if (InputPattern == "Input3") {
            WillReturn.Add("6 1");
            //3 6 5 4 2 1
        }
        else if (InputPattern == "Input4") {
            WillReturn.Add("3 3");
            //2 2 2 1 3 3 3 1 1
        }
        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();
    }

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

        var ValList = new List<long>();
        for (long I = 1; I <= N; I++) {
            for (long J = 1; J <= K; J++) {
                ValList.Add(I);
            }
        }

        var AnswerList = new List<long>();

        // Nが偶数の場合
        if (N % 2 == 0) {
            long FirstVal = N / 2;
            AnswerList.Add(FirstVal);
            ValList.Remove(FirstVal);
            foreach (long EachVal in ValList.OrderByDescending(pX => pX)) {
                AnswerList.Add(EachVal);
            }
        }

        // Nが奇数の場合
        if (N % 2 == 1) {
            long FirstVal = N / 2 + 1;
            for (long I = 1; I <= K; I++) {
                AnswerList.Add(FirstVal);
                ValList.Remove(FirstVal);
            }
            if (ValList.Count > 0) {
                AnswerList.Add(FirstVal - 1);
                ValList.Remove(FirstVal - 1);
                foreach (long EachVal in ValList.OrderByDescending(pX => pX)) {
                    AnswerList.Add(EachVal);
                }
            }
        }

        Console.WriteLine(LongEnumJoin(" ", AnswerList));
    }

    // セパレータとLong型の列挙を引数として、結合したstringを返す
    static string LongEnumJoin(string pSeparater, IEnumerable<long> pEnum)
    {
        string[] StrArr = Array.ConvertAll(pEnum.ToArray(), pX => pX.ToString());
        return string.Join(pSeparater, StrArr);
    }
}

解説

Nの偶奇で場合分けして考えます。

Nの偶奇は、N=6で考えます。
1文字目が1
1文字目が2
1文字目が3
1文字目が4
1文字目が5
1文字目が6

中央値の定義により、1文字目が3での辞書順最大なので、
残りの数値を降順に並べれば良いです。

Nの偶奇は、N=5で考えます。
1文字目が1
1文字目が2
1文字目が3
1文字目が4
1文字目が5

1文字目が3の数列のどれかが解だと分かります。
1文字目が3として、次の文字も、その次の文字も、3が使えれば、使うのが中央値の条件となります。
3を使い終わったら、使用可能文字が偶数になるので、
上記の偶数の場合に帰着して解くことができます。