1287: [蓝桥杯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

public static void main(String[] args){
    	fun();	
    }
    
    public static void fun() {
    	Scanner sc = new Scanner(System.in);
    	while(sc.hasNext()) {
    		int m = sc.nextInt();
    		int x = (int)Math.sqrt(m);
    		boolean flag = false;
    		for(int a = 0; a <= x; a++) {
    			for(int b = a; b <= x; b++) {
    				if(a*a + b*b < m) {
    					for(int c = b; c <=x; c++) {
    						if(a*a + b*b + c*c < m) {
    							int dd = m - a*a -b*b -c*c;
    							int d = (int)Math.sqrt(dd);
    							if(a*a+b*b+c*c+d*d == m) {
    								System.out.println(a+" "+b+" "+c+" "+d);
    								flag = true;
    								break;
    							}
    						}
    						if(flag) break;
    					}
    				}
    				if(flag) break;
    			}
    			if(flag) break;
    		}	
    	}
    }