16年蓝桥杯第八题-四平方和

四平方和
四平方和定理，又称为拉格朗日定理：
每个正整数都可以表示为至多4个正整数的平方和。
如果把0包括进去，就正好可以表示为4个数的平方和。

比如：
5 = 0^2 + 0^2 + 1^2 + 2^2
7 = 1^2 + 1^2 + 1^2 + 2^2
（^符号表示乘方的意思）

对于一个给定的正整数，可能存在多种平方和的表示法。
要求你对4个数排序：
0 <= a <= b <= c <= d
并对所有的可能表示法按 a,b,c,d 为联合主键升序排列，最后输出第一个表示法

程序输入为一个正整数N (N<5000000)
要求输出4个非负整数，按从小到大排序，中间用空格分开

例如，输入：
5
则程序应该输出：
0 0 1 2
再例如，输入：
12
则程序应该输出：
0 2 2 2
再例如，输入：
773535
则程序应该输出：
1 1 267 838

资源约定：
峰值内存消耗 < 256M
CPU消耗 < 3000ms

请严格按要求输出，不要画蛇添足地打印类似：“请您输入...” 的多余内容。

所有代码放在同一个源文件中，调试通过后，拷贝提交该源码。

注意: main函数需要返回0
注意: 只使用ANSI C/ANSI C++ 标准，不要调用依赖于编译环境或操作系统的特殊函数。
注意: 所有依赖的函数必须明确地在源文件中 #include <xxx>， 不能通过工程设置而省略常用头文件。

提交时，注意选择所期望的编译器类型。

解题思路：建立一个结构体数组，用来存储前两位和他们的平方和，接下来我们用二重循环来确定后两位，确定后两位后，求出num-后两位平方和，并在结构体数组中匹配，匹配成功后输出。注意这里有个筛选条件，第一位到第四位是依次增加或相等的。

#include <stdio.h>
#include <math.h>
#include <malloc.h>
typedef struct{
	int i;
	int j;
	int sum;
}*pRecode,recode;
#define LENGTH (pRecode)malloc(sizeof(recode)*num)
pRecode recodeNum;  //全局变量 recodeNum 
int count=0;

void init(int num);
int search(int sum);
void solve(int num);

int main(void)
{
	int num=0;
	scanf("%d",&num);
	solve(num);
	return 0;
}

void solve(int num)
{
	int i=0,j=0,k=0,sum=0;
	int tmp = (int)sqrt(num);
	init(num);   
	for(i=0;i<=tmp;i++)
	{
		for(j=i;j<=tmp;j++)
		{
			sum = num-i*i-j*j;
			k = search(sum);
			if(k>=0&&recodeNum[k].j<=i)
			{
				printf("%d = %d^2 + %d^2 + %d^2 + %d^2",num,recodeNum[k].i,recodeNum[k].j,i,j);
				return;
			}
		}
	}
}
//搜索对应的sum值在recodeNum中的位置，并返回 
int search(int sum) 
{
	int i=0;
	for(i=0;i<count;i++)
	{
		if(recodeNum[i].sum==sum) return i;
	}
	return -1;
}
//根据num值初始化recodeNum 
void init(int num)
{
	int i=0,j=0;
	int tmp = (int)sqrt(num);
	recodeNum = LENGTH;
	if(recodeNum!=NULL)
	{
		for(i=0;i<=tmp;i++)
		{
			for(j=i;j<=tmp;j++)
			{
				int sum=i*i+j*j;
				if(sum<=num)
				{
					recodeNum[count].i = i;
					recodeNum[count].j = j;
					recodeNum[count].sum = sum;
					count++;
				}
			}
		}
	}
}