結果

問題 No.800 四平方定理
コンテスト
ユーザー 👑 obakyan
提出日時 2019-06-06 22:43:36
言語 Lua
(LuaJit 2.1.1734355927)
結果
AC  
実行時間 124 ms / 2,000 ms
コード長 536 bytes
コンパイル時間 242 ms
コンパイル使用メモリ 7,076 KB
実行使用メモリ 68,232 KB
最終ジャッジ日時 2024-10-01 20:24:28
合計ジャッジ時間 2,825 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

local mfl, mce = math.floor, math.ceil
local mmi = math.min

local n, d = io.read("*n", "*n")
local sq = {}
local box = {}
for i = 1, 2 * n * n do box[i] = 0 end
for i = 1, n do
  sq[i] = i * i
  box[2 * sq[i]] = 1
end
for i = 1, n - 1 do
  for j = i + 1, n do
    box[sq[i] + sq[j]] = box[sq[i] + sq[j]] + 2
  end
end
local cnt = 0
for w = 1, n do
  local w2d = d + sq[w]
  for z = 1, n do
    local tgt = w2d - sq[z]
    if tgt <= 0 then break
    elseif tgt <= 2 * n * n  then
      cnt = cnt + box[tgt]
    end
  end
end
print(cnt)
0