java求一个数的四平方和

题目:

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

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

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

比如：

5 = 0^2 + 0^2 + 1^2 + 2^2

7 = 1^2 + 1^2 + 1^2 + 2^2

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

要求你对4个数排序：

0 <= a <= b <= c <= d

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

(联合主键：

主键的唯一性是指主键的值不能重复

联合主键的意思就是：使用两个（或者多个）字段的组合来作为主键，比如：（stuid，courseid）这两个组合为主键，（1,2）和（1，3）这两个是可以共存的，但是（1,2）和（1,2）就不符合主键的唯一性了。)

程序输入为一个正整数N (N<5000000)

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

思路：

我们先用scanner类来进行用户输入，然后定义一个int类型的n来接收用户的输入数值，因为我们有至多4个正整数，所以我们定义前三个数为int类型，最后一个数为double类型，因为n的平方根可能为小数，而第四个数我们用n减去前三个数。Math.sqrt()用于返回参数的算数平方根。然后用三个for循环找出前三个数，最后一个数用n减去前三个数便可得到，最后用一个if语句来判断a*a+b*b+c*c+(int)d*d)是否等于用户输入的值n，如果是则输出a，b，c，d

package Lq_算法练习;

import java.util.Scanner;

public class Demo_四平方和 {
    public static void main(String[] args) {
        Scanner sc=new Scanner(System.in);
        System.out.println("请输入一个数：");
        int n=sc.nextInt();
        //前面三个数
        int a,b,c;
        //第四个数
        double d;
        //Math.sqrt()用于返回参数的算数平方根,比如Math.sqrt(4)就是对4进行开方，等于2
        double temp=Math.sqrt(n); // 减少时间复杂度
        for(a=0;a<temp;a++) {
            for(b=0;b<temp;b++) {
                for(c=0;c<temp;c++) {
                    //这里第四层循环省略掉，可以优化程序
                    d=Math.sqrt(n-a*a-b*b-c*c);
                    if((a*a+b*b+c*c+(int)d*d)==n) {
                        System.out.println("a="+a+" "+"b="+b+" "+"c="+c+" "+"d="+(int)d);
                        return;
                    }
                }
            }
        }
    }
}

运行结果示例：