【蓝桥杯】四平方和

四平方和

题目描述

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

输出格式

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

数据范围

1 <= N <= 5 * 10^6

输入样例:

5

输出样例:

0 0 1 2

代码1 纯暴力做法

分析：问题比较特殊 循环次数没有那么多。经过测试 a和b不会超过十 就找到了答案。

#include<iostream>
#include<cmath>
using namespace std;
int n;
int main() {
	scanf("%d", &n);
	for (int a = 0; a * a <= n; a++) {
		for (int b = a; a * a + b * b <= n; b++) {
			for (int c = b; a * a + b * b + c * c <= n; c++) {
				for (int d = c; a * a + b * b + c * c + d * d <= n; d++) {
					if (a * a + b * b + c * c + d * d == n) {
						cout << a << " " << b << " " << c << " " << d << " ";
						return 0;
					}
				}
			}
		}
	}
	return 0;
}

代码2 正常一点 二分

正常来说，应该是可以随机模拟两个数，所以采用空间换时间的策略，首先将$c^2+d^2$记录下来，然后枚举 a和b

#include<iostream>
#include<cmath>
#include<algorithm>
using namespace std;
int n;
const int N = 5000000;
struct sum {
	int s, c, d;
	bool operator<(const sum& t) {
		if (s != t.s) return s < t.s;
		if (c != t.c) return c < t.c;
		if (d != t.d) return d < t.d;
	}
}h[N];
int idx = 0;
int main() {
	scanf("%d", &n);
	for (int c = 0; c * c <= n; c++) {
		for (int d = c; c * c + d * d <= n; d++) {
			h[idx++] = { c * c + d * d,c,d };
		}
	}
	sort(h, h + idx);
	for (int a = 0; a * a <= n; a++) {
		for (int b = a; a * a + b * b <= n; b++) {
			int l = 0, r = idx - 1;
			while (l < r) {
				int mid = l + r >> 1;
				if (h[mid].s < n - a * a - b * b) l = mid + 1;
				else r = mid;
			}
			if (h[l].s + a * a + b * b == n) {
				cout << a << " " << b << " " << h[l].c << " " << h[l].d << " ";
				return 0;
			}
		}
	}
	return 0;
}

代码3 哈希

#include<iostream>
#include<cmath>
#include<algorithm>
#include<map>
using namespace std;
#define x first
#define y second
typedef pair<int, int> PII;

int n;
map<int, PII> mp;
int idx = 0;
int main() {
	scanf("%d", &n);
	for (int c = 0; c * c <= n; c++) {
		for (int d = c; c * c + d * d <= n; d++) {
			if (!mp.count(c * c + d * d))mp[c * c + d * d] = { c,d };
		}
	}
	for (int a = 0; a * a <= n; a++) {
		for (int b = a; a * a + b * b <= n; b++) {
			if (mp.count(n - a * a - b * b)) {
				cout << a << " " << b << " " << mp[n - a * a - b * b].x << " " << mp[n - a * a - b * b].y << " ";
				return 0;
			}
		}
	}
	return 0;
}

注意

蓝桥杯 不支持 c++11 所以unordered_xx 和auto 不能使用。