蓝桥杯试题：四平方和 java（穷举）

import java.util.Scanner;

/*

练习题：四平方和
四平方和定理，又称为拉格朗日定理：每个正整数都可以表示为至多四个正整数的平方和。如果把 0 包括进去，
就正好可以表示为四个数的平方和。
比如：
5=02+02+12+22
7=12+12+12+22
则对于一个给定的正整数 n，可以表示为：n=a2+b2+c2+d2。
你需要求出 字典序 最小的一组解 a,b,c,d。
字典序大小：从左到右依次比较，如果相同则比较下一项，直到有一项不同，较小的一方字典序更小，反之字典序更大，
所有项均相同则二者字典序相同。
输入格式
程序输入为一个正整数 N(1≤N≤5000000)。
输出格式
输出四个非负整数 a,b,c,d，中间用空格分开。
样例输入1
5
样例输出1
0 0 1 2
样例输入2
12
样例输出2
0 2 2 2

 */
public class FourNumberSum {
 public static void main(String [] args){
     Scanner sc=new Scanner(System.in);
     int n=sc.nextInt();
     for (int i = 0; i <Math.sqrt(n) ; i++) {
         for (int j = 0; j <Math.sqrt(n) ; j++) {
             for (int k = 0; k <Math.sqrt(n) ; k++) {
                 for (int l = 0; l <Math.sqrt(n) ; l++) {
                     if (i*i+j*j+k*k+l*l==n){
                         System.out.println(i+" "+j+" "+k+" "+l);
                         return;
                     }
                 }
             }
         }
     }
 }
}