2016蓝桥杯省赛C/C++A组第八题 四平方和

题意：

四平方和定理，又称为拉格朗日定理： 
每个正整数都可以表示为至多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

分析：直接暴力，三层循环。

#include<bits/stdc++.h>
using namespace std;
int main(){
    int N;
    scanf("%d", &N);
    bool ok = false;
    for(int a = 0; a * a <= N; ++a){
        for(int b = 0; b * b <= N; ++b){
            for(int c = 0; c * c <= N; ++c){
                int tmp = N - a * a - b * b - c * c;
                double t1 = (double)tmp;
                int t = (int)sqrt(t1);
                if(t * t == tmp){
                    printf("%d %d %d %d\n", a, b, c, t);
                    ok = true;
                    break;
                }
            }
            if(ok) break;
        }
        if(ok) break;
    }
    return 0;
}