蓝桥杯——四平方和

#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
#define N 1118*1118+10
int sum[1200];
int vis[N];
int main()
{
	int n;
	cin>>n;
	memset(vis,0,sizeof(vis));
	for(int i=0;i<=1118;i++){
		sum[i]=i*i;
		vis[i*i]=1;
	}
	for(int i=0;i<=1118;i++)
	{
		for(int j=i;j<=1118;j++)
		{
			for(int k=j;k<=1118;k++)
			{
				int tmp = sum[i]+sum[j]+sum[k];
				if((n-tmp)>=0 && (n-tmp)<=1118*1118 && vis[n-tmp]==1){
					cout<<i<<" "<<j<<" "<<k<<" "<<sqrt(n-tmp)<<endl;
					return 0;
				}
			}
		}
	}
	return 0;
}