时间限制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;
}