ABC254-D Together Square

問題へのリンク

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");
            //6
        }
        else if (InputPattern == "Input2") {
            WillReturn.Add("254");
            //896
        }
        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]);

        var HeihousuuSet = new HashSet<long>();
        for (long I = 1; I * I <= N; I++) {
            HeihousuuSet.Add(I * I);
        }

        long[] RevHeihousuuArr = HeihousuuSet.OrderByDescending(pX => pX).ToArray();

        long Answer = 0;
        for (long I = 1; I <= N; I++) {
            long CurrAnswer = 0;

            // 平方数の場合
            if (HeihousuuSet.Contains(I)) {
                CurrAnswer = HeihousuuSet.Count;
            }
            else {
                long Div = I;
                foreach (long EachLong in RevHeihousuuArr) {
                    if (Div % EachLong == 0) {
                        Div /= EachLong;
                        break;
                    }
                }

                foreach (long EachLong in HeihousuuSet) {
                    if (Div * EachLong > N) break;
                    CurrAnswer++;
                }
            }
            Answer += CurrAnswer;
        }
        Console.WriteLine(Answer);
    }
}

解説

最初にN以下の平方数リストをHashSetに用意します。

N以下の数を全探索し、
平方数の場合は、任意の平方数とペアになれます。
平方数でない場合は、なるべく大きい平方数で割った商をDivとし
平方数リストでDiv倍してもN以下な数とペアになれます。