ABC388-E Simultaneous Kagamimochi

問題へのリンク

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("6");
            WillReturn.Add("2 3 4 4 7 10");
            //3
        }
        else if (InputPattern == "Input2") {
            WillReturn.Add("3");
            WillReturn.Add("387 388 389");
            //0
        }
        else if (InputPattern == "Input3") {
            WillReturn.Add("24");
            WillReturn.Add("307 321 330 339 349 392 422 430 477 481 488 537 541 571 575 602 614 660 669 678 712 723 785 792");
            //6
        }
        else {
            string wkStr;
            while ((wkStr = Console.ReadLine()) != null) WillReturn.Add(wkStr);
        }
        return WillReturn;
    }

    static long[] mAArr;
    static long UB;

    static void Main()
    {
        List<string> InputList = GetInputList();
        mAArr = InputList[1].Split(' ').Select(pX => long.Parse(pX)).ToArray();
        Array.Sort(mAArr);
        UB = mAArr.GetUpperBound(0);

        if (mAArr.Min() * 2 > mAArr.Max()) {
            Console.WriteLine(0);
            return;
        }

        long L = 0;
        long R = mAArr.Length;

        while (L + 1 < R) {
            long Mid = R / 2;
            if (R < long.MaxValue) {
                Mid = (L + R) / 2;
            }

            bool Result = CanAchieve(Mid);
            //Console.WriteLine("CanAchieve({0})={1}", Mid, Result);

            if (Result) {
                L = Mid;
            }
            else {
                R = Mid;
            }
        }
        Console.WriteLine(L);
    }

    // K個作れるかを返す
    static bool CanAchieve(long pK)
    {
        if (pK * 2 > mAArr.Length) {
            return false;
        }

        var List1 = new List<long>();
        for (long I = 0; I <= pK - 1; I++) {
            List1.Add(mAArr[I]);
        }

        var List2 = new List<long>();
        for (long I = UB - pK + 1; I <= UB; I++) {
            List2.Add(mAArr[I]);
        }

        for (int I = 0; I <= List1.Count - 1; I++) {
            long A = List1[I];
            long B = List2[I];

            if (2 * A <= B) {
                continue;
            }
            return false;
        }
        return true;
    }
}

解説

作る個数を二分探索してます。

例えば、3個作るなら
最小の3個と最大の3個で試すのが最適です。

試し方としては、
最小の3個が3 4 5
最大の3個がA B C
なら
3 と A
4 と B
5 と C
で実現不可なら、他の組み合わせでも実現不可なので
1通りを試せば十分です。