原題頁面

CSES 3355 - Sum of Four Squares

1.0s 512M

根據 Lagrange 的四平方和定理，任一非負整數都可以表示為四個平方數的和。你的任務是對於一個給定的數字，找出這樣的一組表示法。

輸入格式

第一行包含一個整數 \(t\)：測試案例的組數。

接下來 \(t\) 行，每行包含一個整數 \(n\)。

輸出格式

對每個測試案例輸出四個非負整數 \(a\)、\(b\)、\(c\)、\(d\) 使得 \(n = a^2 + b^2 + c^2 + d^2\)。

限制

  • \(1 \le t \le 1000\)
  • \(0 \le n \le 10^7\)
  • 所有測試中 \(n\) 的總和至多為 \(10^7\)

範例輸入 1

3
5
30
322266

範例輸出 1

2 1 0 0
1 2 3 4
314 159 265 358

題目來源

題目來自 CSES Problem Set（Antti Laaksonen），授權 CC BY-NC-SA 4.0；本頁為翻譯，以相同授權分享。

題目頁說明

快速鍵

主要功能

  • 範例測試 — 執行題目附帶的範例測資並自動比對預期輸出。
  • 自訂測試 — 自己貼 stdin 執行程式。可勾選「與預期輸出比對 (diff)」做行對行比對。
  • 模板 — 貼上你在個人資料設定的預設程式碼模板。
  • 協作 — 與其他同學共筆編輯這題的程式碼。
  • 自動草稿 — 編輯器內容每 1.5 秒自動存到瀏覽器（per 帳號 / 題目 / 語言）。
  • 提交 — 把程式碼交給 judge 評測，回傳 AC / WA / TLE 等結果。

限制

  • 程式碼最多 65,536 字元
  • 自訂測試 stdin 與預期輸出各最多 1 MB （約 100 萬字元）
  • 自訂測試與範例測試共用一個沙箱，每人約 3 秒 1 次 （範例測試 1 秒 1 次）
  • 自訂測試與範例測試都有 15 秒 牆鐘上限（正式評測仍依題目原本時限）
  • 互動題不提供自訂測試（無法模擬與 judge 互動）。
  • 提交評測本身沒有 rate limit，但同題短時間內多次提交會被視為刷分。