1287: [蓝桥杯2016初赛]四平方和 C/C++

题目描述：

四平方和定理，又称为拉格朗日定理：每个正整数都可以表示为至多4个正整数的平方和。
如果把0包括进去，就正好可以表示为4个数的平方和。
比如：
5 = 0^2 + 0^2 + 1^2 + 2^2
7 = 1^2 + 1^2 + 1^2 + 2^2（^符号表示乘方的意思）
对于一个给定的正整数N，可能存在多种平方和的表示法。
要求你对4个数排序：0 <= a <= b <= c <= d
并对所有的可能表示法按 a,b,c,d 为联合主键升序排列，最后输出第一个表示法

输入格式：

输入存在多组测试数据，每组测试数据输入一行为一个正整数N (N<5000000)

输出格式：

对于每组测试数据，要求输出4个非负整数，按从小到大排序，中间用空格分开

输入样例：

5
12
773535

输出样例：

0 0 1 2
0 2 2 2
1 1 267 838

 首先想到的肯定是用四层for的暴力解法，结果会超时，因此在此基础上做一定的优化。不难发现其实只用三层for就可以解决，因为确定了前三个数最后一个数可以直接算出来，同时前三个数的搜索范围也可以缩小。另外题目要求输入多组测试数据，并且每组只输出第一个答案，所以当找到每组的第一个答案时，需要一个跳出三层for循环的操作，因为break只能跳出当前循环，所以这里用到了一个goto函数。

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

int main()
{
	int N;
	mark:
	while(scanf("%d",&N) != EOF)
	{
		int a,b,c,d;
		for(a = 0;a <= sqrt(N/4);a++)
		{
			for(b = a;b <= sqrt(N/3);b++)
			{
				for(c = b;c <= sqrt(N/2);c++)
				{
					int d =sqrt(N - (a*a + b*b + c*c));
					if(a*a + b*b + c*c+ d*d == N)
					{		
						cout << a << " " << b << " " << c << " " << d << endl;									
						goto mark;
 					}
				}
			}
		}
	} 
	
	return 0;
}