计蒜客四平方和

#include<iostream>
#include<cstdlib>
#include<cmath>
#include<cstdio>
using namespace std;

int main()
{
    //freopen("1.txt","r",stdin);
    //freopen("2.txt","w",stdout);
    int n;
    bool finish=true;
    int z;
    int s;
    
    cin>>n;
    int m=sqrt(n);
    if(m*m!=n)m++;

    for(int i=0;i<m;i++){
        if(finish)
            for(int j=0;j<m;j++){
                if(finish)
                    for(int k=0;k<=m;k++)
                    if(finish){
                        z=k*k+j*j+i*i;
                        s=sqrt(n-z);
                        if(s*s==n-z){
                            finish=false;
                            printf("%d %d %d %d",i,j,k,s);
                        }
                    }
            }
    }
    return 0;
}