题目描述
四方定理是数论中著名的一个定理,指任意一个自然数都可以拆成四个自然数的平方之和。例如:
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≤1,000,000
输入样例
25
输出样例
0 0 0 5
0 0 3 4
1 2 2 4
思路解析
用多重循环后sort。
AC代码
#include<bits/stdc++.h>
using namespace std;
int main() {
int n;
cin>>n;
vector<vector<int>> s;
for(int i=0;i*i<=n;i++) {
for(int j=i;j*j<=n-j*j;j++) {
for(int k=j;k*k<=n-i*i-j*j;k++) {
int l_sqrt=n-i*i-j*j-k*k;
int l=static_cast<int>(sqrt(l_sqrt));
if(l_sqrt==l*l&&l>=k) {
s.push_back({i,j,k,l});
}
}
}
}
sort(s.begin(),s.end());
s.erase(unique(s.begin(),s.end()),s.end());
for(const vector<int>& sol : s) {
for(int val : sol) {
cout<<val<<" ";
}
cout<<"\n";
}
return 0;
}