WUSTOJ 1338: The minimum square sum（Java）

题目链接：1338: The minimum square sum

Description

Given a prime p(p<10⁸), you are to find min{x²+y²}, where x and y belongs to positive integer, so that x²+y²=0 (mod p).

输入一个质数 p，你找出两个正整数 x 和 y 使得 (x²+y²) mod p = 0，且 x²+y² 最小。

Input

Every line is a p. No more than 10001 test cases.

Output

The minimum square sum as described above.

Sample Input

2
3
5
7
11
13

Sample Output

2
18
5
98
242
13

分析?

没什么好分析的，就是找规律，动手算几个就行了。。。

OJ题目显示有点小问题QAQ

代码?

/**
 * Time 1412ms
 * @author wowpH
 * @version 3.1
 * @date 2019年7月2日下午5:04:43
 * Environment:	Windows 10
 * IDE Version:	Eclipse 2019-3
 * JDK Version:	JDK1.8.0_112
 */

import java.util.Scanner;

public class Main {
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		while (sc.hasNext()) {
			long prime = sc.nextLong();
			if (2 == prime) {
				System.out.println(2);
			} else if (0 == (prime - 3) % 4) {
				System.out.println(2 * prime * prime);
			} else {
				System.out.println(prime);
			}
		}
		sc.close();
	}
}