藍橋杯 四數平方和
四平方和定理,又稱爲拉格朗日定理:
每個正整數都可以表示爲至多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
輸入樣例:
在這裏給出一組輸入。例如:
773535
輸出樣例:
在這裏給出相應的輸出。例如:
1 1 267 838
思路:
這道題可以直接採取暴力求解,用三重循環硬算,在每層循環中加入一些限定條件,每次判斷當前層的平方是否已經超過了x,超過了直接break即可,可以節省一下時間。
代碼:
#include<iostream>
#include<cmath>
#include<algorithm>
using namespace std;
int main()
{
int x,t,l;
cin >> x;
for (int i = 0; i < 3000; i++)
{
for (int j = 0; j < 3000; j++)
{
if (i*i + j*j>x)
break;
for (int k = 0; k < 3000; k++)
{
t = x - i*i - j*j - k*k;
l = sqrt(t);
if (l*l == t)
{
cout << i << ' ' << j << ' ' << k << ' ' << l << endl;
return 0;
}
if (i*i + j*j + k*k+l*l>x)
break;
}
}
}
}