题目描述
四方定理是数论中著名的一个定理,指任意一个自然数都可以拆成四个自然数的平方之和。例如:
$$25=1^2 + 2^2 + 2^2 + 4^2$$
对25来说,还有其他方案:
$$25=0^2 + 0^2 + 3^2 + 4^2$$
以及
$$25=0^2 + 0^2 + 0^2 + 5^2$$
给定一个自然数 $n$,请输出 $n$ 的所有四平方拆分方案。
输入描述
单个整数:表示 $n$。
数据范围
0≤n≤50,000。
输出描述
若干行:每行四个由小到大排列的自然数,表示一种拆分方案。如果有多种方案,先输出首项较小的方案,对于首项相同的方案,先输出第二项较小的方案,其他情况以此类推。
输入样例
25
输出样例
0 0 0 5
0 0 3 4
1 2 2 4
#include <iostream>
#include <cmath>
using namespace std;
long long n, i, j, k, l;
int main() {
cin.tie(0);
cin >> n;
double f = int(sqrt(n));
for (i = 0; i <= f; ++i)
for (j = i; j <= f; ++j)
for (k = j; k <= f; ++k)
for (l = k; l <= f; ++l)
if ((i * i + j * j + k * k + l * l) == n)
cout << i << " " << j << " " << k << " " << l << endl;
return 0;
}