ARC115-C N Coloring

問題へのリンク

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");
            //1 2 2 3
        }
        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[] AnswerArr = new long[N + 1];
        for (long I = 1; I <= N; I++) {
            long[] YakusuuArr = DeriveYakusuuArr(I);
            var KouhoList = new List<long>();

            Array.ForEach(YakusuuArr, pX => KouhoList.Add(AnswerArr[pX]));
            AnswerArr[I] = KouhoList.Max() + 1;
        }

        var sb = new System.Text.StringBuilder();
        for (int I = 1; I <= N; I++) {
            sb.Append(AnswerArr[I]);
            if (I < N) sb.Append(" ");
        }
        Console.WriteLine(sb.ToString());
    }

    // 約数を列挙する
    static long[] DeriveYakusuuArr(long pTarget)
    {
        var YakusuuSet = new HashSet<long>();
        for (long I = 1; I * I <= pTarget; I++) {
            if (pTarget % I == 0) {
                YakusuuSet.Add(I);
                YakusuuSet.Add(pTarget / I);
            }
        }
        long[] YakusuuArr = YakusuuSet.ToArray();
        Array.Sort(YakusuuArr);
        return YakusuuArr;
    }
}

解説

小さい数から順に埋めていくと
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
1 2 2 3 2 3 2 4 3  3  2  4  2  3  3
になって、(約数の中での最大値 + 1) をナイーブに求めていけば良いです。