蓝桥杯历届真题-四平方和-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 为联合主键升序排列，最后输出第一个表示法

输入格式:

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

输出格式:

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

输入样例1：

5

输出样例1：

0 0 1 2

输入样例2：

773535

输出样例2：

1 1 267 838

思路：

这个题思路很简单，只需要用三重循环遍历，求差后开方求第四个数并对第四个数进行判断，符合即输出结束，不符合继续遍历。

代码如下附上：

import java.beans.beancontext.BeanContext;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.math.BigDecimal;
import java.math.BigInteger;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Calendar;
import java.util.Collection;
import java.util.Collections;
import java.util.Comparator;
import java.util.GregorianCalendar;
import java.util.Scanner;
import java.util.function.IntToDoubleFunction;

import javax.print.attribute.Size2DSyntax;
import javax.security.auth.x500.X500Principal;
import javax.swing.text.Highlighter;
import javax.xml.crypto.Data;

import org.omg.PortableServer.ID_ASSIGNMENT_POLICY_ID;


public class Main{
	public static void main(String args[]) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		Scanner sc = new Scanner(System.in);
		
		int a, b, c;
		a = Integer.parseInt(br.readLine());
		for (int i = 0; i < 3000; i++) {
			for (int j = 0; j < 3000; j++) {
				if(i * i + j * j > a) {
					break;
				}
				for (int j2 = 0; j2 < 3000; j2++) {
					b = a - i * i - j * j - j2 * j2;
					c = (int) Math.sqrt(b);
					if(c * c == b) {
						System.out.println(i + " " + j + " " + j2 + " " + c);
						return;
					}
					if(i * i + j * j + j2 * j2 + c * c > a)
						break;
				}
			}
		}
	}
}