結果

問題 No.800 四平方定理
コンテスト
ユーザー shi-mo
提出日時 2020-09-29 23:27:50
言語 D
(dmd 2.109.1)
結果
AC  
実行時間 111 ms / 2,000 ms
コード長 592 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 1,848 ms
コンパイル使用メモリ 137,940 KB
実行使用メモリ 67,420 KB
最終ジャッジ日時 2024-06-22 09:11:55
合計ジャッジ時間 4,442 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import std.stdio;
import std.conv;
import std.array;

void main() {
    uint n, d;
    readf("%s %s\n", &n, &d);

    immutable uint m = 2*n*n;
    uint[] cxy = new uint[](m+1);
    uint[] cwz = new uint[](m+1);

    foreach (i; 1..(n+1)) {
        uint i2 = i*i;
        foreach (j; 1..(n+1)) {
            uint j2 = j*j;
            cxy[i2 + j2]++;

            uint wz = i2 - j2 + d;
            if (0 < wz && wz <= m) {
                cwz[wz]++;
            }
        }
    }

    uint sum = 0;
    foreach (i; 2..(m+1)) {
        sum += cxy[i] * cwz[i];
    }
    writef("%d\n", sum);
}
0