蓝桥杯2016年第七届真题——四平方和

四平方和

---

一、题目内容

题目描述

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

---

二、思路分析

        暴力+稍微优化就可以AC了，如果用4层for循环直接暴力的话，大概率是会超时的，那么就需要稍微的优化一下。我们会发现最后一层for循环是没必要的，因为它可以通过前三次循环来计算出来的，再进行判断是否符合就可以了，并且用x*x<=n/k来控制循环的终止条件可以减少循环次数。

---

三、代码实现

#include <bits/stdc++.h>
using namespace std;

int main() {
	int n;
	cin >> n;
	for (int a = 0; a * a <= n / 4; a++) { //暴力+稍微优化
		for (int b = a; b * b <= n / 3; b++) {
			for (int c = b; c * c <= n / 2; c++) {
				int d = n - a * a - b * b - c * c;
				if (floor(sqrt(d) + 0.5) == sqrt(d)) { //floor函数向下取整 判断sqrt(d)是否为整数
					cout << a << " " << b << " " << c << " " << sqrt(d) << endl;
					return 0; //找到第一个后直接返回退出
				}
			}
		}
	}

	return 0;
}

---

加油哦！ 如有错误和需要改进完善之处，欢迎大家纠正指教。