该题看到时间限制是3s就确定可以使用暴力了
思路分析:
枚举每一个需要的数字,直到找到满足条件的第一个答案输出即可(因为都是从0开始枚举,所以第一个自然是字典序最小的)
#include<iostream>
#include<cmath>
using namespace std;
int main() {
int n;
cin >> n;
int len = sqrt(n);//最大不可能超过len
for (int i = 0; i <= len; i++) {
for (int j = i; j <= len; j++) {
for (int k = j; k <= len; k++) {
for (int z = k; z <= len; z++) {
if (i * i + j * j + k * k + z * z > n)break;//小优化
if (i * i + j * j + k * k + z * z == n) {
cout << i << " " << j << " " << k << " " << z << endl;
exit(0);
}
}
}
}
}
}
思路分析2:
我们可以分成两段进行枚举,先枚举所需的c和d,再以哈希的形式去枚举a和b进行对应,直到找到第一个与哈希对应的a和b(与前面一样因为都是从0开始枚举,所以第一个自然是最小的)
#include<iostream>
using namespace std;
const int MAX = 5e6 + 10;
typedef struct Node {
int v, c, d;
}Arr[MAX];//哈希表
Arr map;
int main() {
int n;
cin >> n;
for (int c = 0; c * c <= n; c++) {
for (int d = 0; c * c + d * d <= n; d++) {//先枚举c和d
int temp = n - c * c - d * d;
if (!map[temp].v) {//未被占用
map[temp].v = 1;
map[temp].c = c;
map[temp].d = d;
}
}
}
for (int a = 0; a*a <= n; a++) {
for (int b = 0; a * a + b * b <= n; b++) {
int temp = a * a + b * b;
if (map[temp].v) {//找到对应
cout << a << " " << b << " " << map[temp].c << " " << map[temp].d << endl;
return 0;
}
}
}
return 0;
}