第七届蓝桥杯JavaB组省赛-四平方和

题目

四平方和定理，又称为拉格朗日定理：
　　每个正整数都可以表示为至多4个正整数的平方和。
　　如果把0包括进去，就正好可以表示为4个数的平方和。

比如：
　　5 = 0² + 0² + 1² + 2²
　　7 = 1² + 1² + 1² + 2²
　　（^符号表示乘方的意思）

对于一个给定的正整数，可能存在多种平方和的表示法。
　　要求你对4个数排序：
　　0 <= a <= b <= c <= d
　　并对所有的可能表示法按 a,b,c,d 为联合主键升序排列，最后输出第一个表示法

程序输入为一个正整数N (N<5000000)
　　要求输出4个非负整数，按从小到大排序，中间用空格分开

例如，输入：
　　5
　　则程序应该输出：
　　0 0 1 2

再例如，输入：
　　12
　　则程序应该输出：
　　0 2 2 2

再例如，输入：
　　773535
　　则程序应该输出：
　　1 1 267 838

资源约定：
　　峰值内存消耗 < 256M
　　CPU消耗 < 3000ms

思路：暴力枚举+空间换取时间优化

这个题暴力枚举可以直接写四个for循环，但是数据规模是(10⁶)⁴,绝对超时。
但是可以先将c²+d²计算出来存起来，然后通过优化，c²<=n/2, d²<=n。
后续只要去哈希表中查找**N-a²-b²**即可。

public class test_01 {
	
	static class E
	{
		int c;
		int d;
		public E(int c,int d) 
		{
			this.c=c;
			this.d=d;
		}
	}
	static HashMap<Integer,E> map=new HashMap<Integer, E>();
    public static void main(String[] args)
    {
    	Scanner in=new Scanner(System.in);
    	int n=in.nextInt();
    	for (int c = 0; c*c<= n/2; c++) 
			for (int d = 0; d*d<=n; d++) 
				if(!map.containsKey(c*c+d*d))
					map.put(c*c+d*d, new E(c, d));
				
    	for (int a = 0; a*a<=n/4; a++) 
    		for (int b = 0; b*b<n/3; b++) 
				if(map.containsKey(n-(a*a+b*b)))
				{
					System.out.println(a+" "+b+" "+map.get(n-(a*a+b*b)).c+" "+map.get(n-(a*a+b*b)).d);
					return;
				}
    }
}