2016-8-四平方和

四平方和

四平方和定理，又称为拉格朗日定理： 每个正整数都可以表示为至多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                                        12                                          773535
则程序应该输出：
0 0 1 2                               0 2 2 2                                   1 1 267 838

思路：暴力枚举+优化   

优化① 缩小变量的取值范围    double tmp = sqrt(n);

       ② 减少变量个数   计算出d的值    int d  =  (int) sqrt( n-a*a-b*b-c*c-d*d );

重点在优化，否则会超时

 

代码

#include <cstdio>
#include <cmath>
#include<iostream>
#define MAXN 2300     //(int)sqrt(5000000)=2236
using namespace std;
//注意范围，题目给的范围枚举到2300就够了
/*
直接枚举从小的开始，枚举到答案就可以return了，
不过注意的是只枚举三个数，最后一个数可以倒着求出来，然后正过来验证，
*/

int main(){
    int n;
    scanf("%d",&n);
    double tmp=sqrt(n);  //tmp<MAXN  这是关键，不要让时间复杂度太大（超时问题）
    
    for(int a=0; a<tmp; a++){
        
        for(int b=a; b<tmp; b++){
        
            for(int c=b; c<tmp; c++){
            	
                int d = (int)sqrt(n-a*a-b*b-c*c);     // sqrt()返回值是double，强制转换为整形  算出第4个数
                if( a*a+b*b+c*c+d*d==n ){
                    printf("%d %d %d %d\n",a,b,c,d);
                    return 0;              
                }
            }
        }
    }
    return 0;
}