長らく競プロ休止していましたが、リハビリ目的もかねて AtCoder ProblemsのBoot camp for Beginners で遊んでいます。いい感じに苦戦したのでメモ。

問題

問題文

整数 $N,K$ が与えられます。 $N$ 以下の正の整数の組 $(a,b,c)$ であって、 $a+b,b+c,c+a$ がすべて $K$ の倍数であるようなものの個数を求めてください。ただし、 $a,b,c$ の順番を入れ替えただけの組も異なるものとして数えます。また、 $a,b,c$ の中に同じものがあっても構いません。

制約

$1 \leq N,K \leq 2×10^{5}$

$N,K$ は整数である

ACしたコード

using System;
using System.Linq;
using System.Collections.Generic;
using static System.Console;
using System.Text;
using System.IO;
using static System.Math;
namespace AtCoder
{
    class Program
    {
        static public long[] Sarray() { return ReadLine().Split().Select(long.Parse).ToArray(); }
        static void Main(string[] args)
        {
            var NK = Sarray();
            var N = NK[0];
            var K = NK[1];
            var ans = 0L;
            ans = (N / K) * (N / K) * (N / K);
            if (K % 2 == 0 && K / 2 <= N)
            {
                var sel = (N - K / 2) / K + 1;
                ans += sel * sel * sel;
            }
            WriteLine(ans);
        }
    }
}