[蓝桥杯2016初赛]四平方和

[蓝桥杯2016初赛]四平方和

题目描述
四平方和定理，又称为拉格朗日定理：每个正整数都可以表示为至多4个正整数的平方和。
如果把0包括进去，就正好可以表示为4个数的平方和。
比如：
5 = 0^2 + 0^2 + 1^2 + 2^2
7 = 1^2 + 1^2 + 1^2 + 2^2（符号表示乘方的意思）
对于一个给定的正整数N，可能存在多种平方和的表示法。
要求你对4个数排序：0 <= a <= b <= c <= d
并对所有的可能表示法按 a,b,c,d 为联合主键升序排列，最后输出第一个表示法
输入
输入存在多组测试数据，每组测试数据输入一行为一个正整数N (N<5000000)
输出
对于每组测试数据，要求输出4个非负整数，按从小到大排序，中间用空格分开
***样例输入 ***
5
12
773535
***样例输出 ***
0 0 1 2
0 2 2 2
1 1 267 838

package pro;

import java.util.Scanner;

public class Main {
	
	public static void main(String[] args) {
		Scanner in = new Scanner(System.in);
		while(in.hasNext()) {
			int flag = 0;
			int n = in.nextInt();
			for(int i = 0; i < Math.sqrt(n); i++) {
				for(int j = i; j < Math.sqrt(n); j++) {
					if(Math.pow(i, 2)+Math.pow(j, 2)>=n)
						continue;
					for(int k = j; k < Math.sqrt(n); k++) {
						int a = (int)(Math.pow(i, 2)+Math.pow(j, 2)+Math.pow(k, 2));
						if(a>=n) {
							continue;
						}
						int l = (int) Math.sqrt(n-a);
							if(a+Math.pow(l, 2)==n) {
								System.out.println(i+" "+j+" "+k+" "+l);
								flag = 1;
								break;
							}
					}
					if(flag==1)
						break;
				}
				if(flag==1)
					break;	
			}
		}
		
	}
}

暴力枚举,对第四个数进行优化处理

2021-01-30