四平方和 C

Problem Description

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

比如： 5 = 0^2 + 0^2 + 1^2 + 2^25=0​2​​+0​2​​+1​2​​+2​2​​

            7 = 1^2 + 1^2 + 1^2 + 2^27=1​2​​+1​2​​+1​2​​+2​2​​

则对于一个给定的正整数 n，可以表示为：n = a^2 + b^2 + c^2 + d^2​​。

你需要求出字典序最小的一组解 a,b,c,d。

字典序大小：从左到右依次比较，如果相同则比较下一项，直到有一项不同，较小的一方字典序更小，反之字典序更大，所有项均相同则二者字典序相同。

Input

程序输入为一个正整数 N(0<N <5000001)。

Output

输出 4 个非负整数 a,b,c,d 中间用空格分开。。

样例输入

Sample Input

5
12

Sample Output

0 0 1 2
0 2 2 2

 

#include<stdio.h>
#include<math.h>
int main()
{
    int n,a,b,c,d,t;
    scanf("%d",&n);
    for(a=0; a<2500; ++a)
    {
        for(b=a; b<2500; ++b)
        {
            for(c=b; c<2500; ++c)
            {
                d=sqrt(n-a*a-b*b-c*c);
                if(n==a*a+b*b+c*c+d*d)
                {
                    if(c>d)
                    {
                        t=d;
                        d=c;
                        c=t;
                        printf("%d %d %d %d\n",a,b,c,d);
                        return 0;
                    }
                }
            }
        }
    }
    return 0;
}

看了那么久终于用到了暴力枚举的方法~总体的思想就是用三个循环嵌套，最后一个数用已知的三个数表达出来，以此来判断出循环的条件，在排序时啊，a,b,c已经排好顺序了，只需要判断一下b,c的关系。

但是我已经找不到这个题了，只是在CB里测试数据都对，所以不知道超不超时，2500的边界条件是由2500的平方大于5000000得出的，在输出之后加个return能退出这个函数