洛谷：P8635 [蓝桥杯 2016 省 AB] 四平方和

时间限制3.00s         内存限制256.00MB         难易度：普及−

【题目描述】

四平方和定理，又称为拉格朗日定理：

每个正整数都可以表示为至多 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<5×10^6)。

【输出格式】

要求输出 4 个非负整数，按从小到大排序，中间用空格分开。

【输入输出样例】

输入 #1

5

输出 #1

0 0 1 2

输入 #2

12

输出 #2

0 2 2 2

输入 #3

773535

输出 #3

1 1 267 838

【说明/提示】

时限 3 秒, 256M。蓝桥杯 2016 年第七届省赛

蓝桥杯 2016 年省赛 A 组 H 题（B 组 H 题）。

【算法分析】

这道题我们可以用暴力枚举的方法来做。循环遍历前三个数。特殊的，为了节省时间复杂度，防止TLE，我们可以先设t=n-a*a-b*b-c*c，然后将第四个数设为 t 的开方，如果第四个数的平方=t，证明t 可以开出整数的方（即求出了第四个数），将四个数输出后退出循环即可。

【参考代码】

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