蓝桥杯 四平方和【枚举】【哈希表】

四平方和定理，又称为拉格朗日定理：

每个正整数都可以表示为至多 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个非负整数，按从小到大排序，中间用空格分开。

数据范围

0<N<5∗106

输入样例：

5

输出样例：

0 0 1 2

程序说明：

暴力法：三重循环枚举a、b、c，然后判断d是否满足条件

哈希表做法：先枚举c和d，将c * c + d * d 的结果用哈希表存起来。然后枚举a和b，判断能否在哈希表中找到满足条件的c * c + d * d。注意哈希表中只存储第一次出现的结果，这样才能保证哈希表中存储的c最小，从而保证字典序最小。

代码如下：

//暴力枚举
import java.io.*;
import java.util.*;

public class Main {
	static int n;
	
	public static void main(String[] args) {
		Scanner sc = new Scanner(new BufferedInputStream(System.in));
		n = sc.nextInt();
		
		for(int a = 0; a * a <= n; a++)
			for(int b = a; b * b <= n; b++)
				for(int c = b; c * c <= n; c++) {
					int d = (int)Math.sqrt(n - a * a - b * b - c * c);
					if(a * a + b * b + c * c + d * d == n) {
						System.out.println(a + " " + b + " " + c + " " + d);
						return;
					}
				}
	}
}

//哈希表
import java.io.*;
import java.util.*;

public class Main {
	static int n;
	
	public static void main(String[] args) {
		Scanner sc = new Scanner(new BufferedInputStream(System.in));
		n = sc.nextInt();
		Map<Integer, Integer> m = new HashMap<>();
		for(int c = 0; c * c <= n; c++)
			for(int d = c; d * d <= n; d++)
				if(m.get(c * c + d * d) == null)
					m.put(c * c + d * d, c);
		
		for(int a = 0; a * a <= n; a++)
			for(int b = 0; b * b <= n; b++) {
				int t = n - a * a - b * b;
				if(m.get(t) != null) {
					int c = m.get(t);
					System.out.println(a + " " + b + " " + c + " " + (int)(Math.sqrt(n - a * a - b * b - c * c)));
					return;
				}
			}
	}
}