AtCoderのABC    次のABCの問題へ    前のABCの問題へ

ABC268-C Chinese Restaurant


問題へのリンク


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

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

        // 得点[基点の値]なDict
        var ScoreDict = new Dictionary<long, long>();

        for (long I = 0; I <= pArr.GetUpperBound(0); I++) {
            var KouhoList = new List<long>();

            long Mod1 = I - pArr[I];
            Mod1 %= N;
            if (Mod1 < 0) Mod1 += N;

            long Mod2 = I - pArr[I] - 1;
            Mod2 %= N;
            if (Mod2 < 0) Mod2 += N;

            long Mod3 = I - pArr[I] + 1;
            Mod3 %= N;
            if (Mod3 < 0) Mod3 += N;

            KouhoList.Add(Mod1);
            KouhoList.Add(Mod2);
            KouhoList.Add(Mod3);

            foreach (long EachKouho in KouhoList) {
                if (ScoreDict.ContainsKey(EachKouho) == false) {
                    ScoreDict[EachKouho] = 0;
                }
                ScoreDict[EachKouho]++;
            }
        }
        Console.WriteLine(ScoreDict.Values.Max());
    }
}


解説

得点[基点の値]なDictを作って
基点の値がどの添字の場合に加点されるかを集計してます。

例えば、
添字 0 1 2 3 4 5 6 7 8 9
値   3 9 6 1 7 2 8 0 5 4
だと値5は
基点が2か3か4だと加点されます。