No.370 道路の掃除

■■■問題■■■

今日は学校の校外清掃日で、学校の近くの道路に落ちているごみを一人、
N個以上拾うことになっています。

健太君は最初、位置0にいます。
M個のごみの位置Diが与えられるので、
N個のごみを拾う最短の移動距離を求めなさい。

■■■入力■■■

N M

D1
D2
・
・
・
Dm

1 <= N <= M <= 1000
-1万 <= Di <= 1万

i != j のとき、Di != Dj です。

■■■出力■■■

健太君がN個のごみを拾うためにかかる
最短の移動距離を出力してください。 

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("3 5");
            WillReturn.Add("1");
            WillReturn.Add("2");
            WillReturn.Add("3");
            WillReturn.Add("4");
            WillReturn.Add("5");
            //3
            //1,2,3の位置のごみを順に拾います
        }
        else if (InputPattern == "Input2") {
            WillReturn.Add("2 4");
            WillReturn.Add("-5");
            WillReturn.Add("-1");
            WillReturn.Add("2");
            WillReturn.Add("5");
            //4
            //-1の位置のゴミをまず拾ってから、
            //折り返して2の位置のごみを拾うのが最短です
        }
        else if (InputPattern == "Input3") {
            WillReturn.Add("2 5");
            WillReturn.Add("-3");
            WillReturn.Add("-1");
            WillReturn.Add("0");
            WillReturn.Add("2");
            WillReturn.Add("4");
            //1
            //初期位置0にごみがあることもあります。
            //この場合は移動なしで拾えるため、
            //次に-1のごみを拾って終了です。 
        }
        else {
            string wkStr;
            while ((wkStr = Console.ReadLine()) != null) WillReturn.Add(wkStr);
        }
        return WillReturn;
    }

    static void Main()
    {
        List<string> InputList = GetInputList();

        int[] wkArr = { };
        Action<string> SplitAct = pStr =>
            wkArr = pStr.Split(' ').Select(X => int.Parse(X)).ToArray();

        SplitAct(InputList[0]);
        int N = wkArr[0];

        int[] DArr = InputList.Skip(1).Select(X => int.Parse(X)).ToArray();
        Array.Sort(DArr);
        int UB = DArr.GetUpperBound(0);

        int Answer = int.MaxValue;
        for (int I = 0; I <= UB; I++) {
            int MinInd = I;
            int MaxInd = I + N - 1;

            if (MaxInd > UB) break;

            int MinD = DArr[MinInd];
            int MaxD = DArr[MaxInd];

            int KyoriSum = 0;
            if (MinD > 0) { //両方とも正の場合
                KyoriSum = MaxD;
            }
            else if (MaxD < 0) { //両方とも負の場合
                KyoriSum = Math.Abs(MinD);
            }
            else { //符号が異なる場合
                int wkSum1 = Math.Abs(MinD) * 2 + MaxD;
                int wkSum2 = Math.Abs(MinD) + MaxD * 2;

                KyoriSum = Math.Min(wkSum1, wkSum2);
            }
            if (Answer > KyoriSum)
                Answer = KyoriSum;
        }
        Console.WriteLine(Answer);
    }
}

解説

拾うゴミの範囲でループさせ、
両端のゴミの座標の正負で場合分けしてます。