蓝桥杯真题-四平方和第一字典序

import java.util.Scanner;

public class LANQIAO {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int d;
        for (int a = 0; a * a <= n; a++) {
            for (int b = a;a * a + b * b <= n ; b++) {
                for (int c = b; a * a + b * b + c * c <= n; c++) {
                    d = (int) Math.sqrt(n - (a * a + b * b + c * c));
                    // 字典序，所以找到第一种方案直接return
                    if(a * a + b * b + c * c + d * d == n)
                    { System.out.println(a + " "+b + " "+c + " "+d);return;}
                }
            }
        }
    }
}