四方定理!!!

Hello!我是love you 的小羊生煎（>-<)

通过我分享的实用技巧和策略，你将在你的领域脱颖而出，引领潮流！

无论你遇到什么挑战，我将一直在你身边，为你提供支持和鼓励！

2话不说上代码

说明

四方定理是数论中著名的一个定理，指任意一个自然数都可以拆成四个自然数的平方之和。例如：

25 = 1^2 + 2^2 + 2^2 + 4^225=12+22+22+42

对25来说，还有其他方案：

25 = 0^2 + 0^2 + 3^2 + 4^225=02+02+32+42

以及

25 = 0^2 + 0^2 + 0^2 + 5^225=02+02+02+52

给定一个自然数n，请输出n的所有四平方拆分方案。

输入格式

单个整数：表示n。

输出格式

若干行：每行四个由小到大排列的自然数，表示一种拆分方案。如果有多种方案，先输出首项较小的方案，对于首项相同的方案，先输出第二项较小的方案，其他情况以此类推。

提示

0≤n≤50,0000≤n≤50,000

四方定理

#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;
}

输入数据 1

25

Copy

输出数据 1

0 0 0 5
0 0 3 4
1 2 2 4