四平方和 JAVA题解

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

每个正整数都可以表示为至多 44 个正整数的平方和。

如果把 00 包括进去，就正好可以表示为 44 个数的平方和。

比如：

5=02+02+12+225=02+02+12+22
7=12+12+12+227=12+12+12+22

对于一个给定的正整数，可能存在多种平方和的表示法。

要求你对 44 个数排序：

0≤a≤b≤c≤d0≤a≤b≤c≤d

并对所有的可能表示法按 a,b,c,da,b,c,d 为联合主键升序排列，最后输出第一个表示法。

输入格式

输入一个正整数 NN。

输出格式

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

数据范围

0<N<5∗1060<N<5∗106

输入样例：

5

输出样例：

0 0 1 2

import java.util.*;
public class Main{
    public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt();
		for(int a = 0;a <= Math.sqrt(n);a ++) {
			for(int b = a;b <= Math.sqrt(n - a * a) ;b ++) {
				for(int c = b;c <= Math.sqrt(n - a*a - b*b);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;
					}
				}
			}
		}
	}
}