ABC169-D Div Game

問題へのリンク

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("24");
            //3
        }
        else if (InputPattern == "Input2") {
            WillReturn.Add("1");
            //0
        }
        else if (InputPattern == "Input3") {
            WillReturn.Add("64");
            //3
        }
        else if (InputPattern == "Input4") {
            WillReturn.Add("1000000007");
            //1
        }
        else if (InputPattern == "Input5") {
            WillReturn.Add("997764507000");
            //7
        }
        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]);

        if (N == 1) {
            Console.WriteLine(0);
            return;
        }

        Dictionary<long, long> SoinsuuDict = DeriveSoinsuuDict(N);

        long Answer = 0;
        for (long I = 1; I <= SoinsuuDict.Values.Max(); I++) {
            foreach (long EachKey in SoinsuuDict.Keys.ToArray()) {
                if (SoinsuuDict[EachKey] >= I) {
                    SoinsuuDict[EachKey] -= I;
                    Answer++;
                }
            }
        }
        Console.WriteLine(Answer);
    }

    // 素因数分解し、指数[素数]なDictを返す
    static Dictionary<long, long> DeriveSoinsuuDict(long pTarget)
    {
        var WillReturn = new Dictionary<long, long>();

        long CurrVal = pTarget;
        // ルートより大きい数を、素因素に持つとしても、1つだけのため
        // ルートまで試し割りを行い、素因数が残っていたら、追加する
        for (long I = 2; I * I <= pTarget; I++) {
            if (CurrVal % I > 0) continue;

            WillReturn[I] = 0;
            while (CurrVal % I == 0) {
                WillReturn[I]++;
                CurrVal /= I;
            }
            if (CurrVal == 1) break;
        }

        // 素因数が残っている場合
        if (CurrVal > 1) {
            WillReturn[CurrVal] = 1;
        }
        return WillReturn;
    }
}

解説

Nを、素べきで最大で何回割れるかを求めるので
Nを素因数分解して、素数ごとに
1乗で割れるか、2乗で割れるか、3乗で割れるか、をチェックしてます。

素因数分解に関しては、
たとえば、106という合成数を素因数分解すると
106 = 2 * 53
ですが、106の平方根は、10.29で
平方根より大きい数を素因数に持つとしても、1つだけとなります。

なので、素因数分解では、
平方根まで試し割りを行い、素因数が残っていたら、追加する
といった方法で計算量をルート(N)まで落とせます。