Neliöt

import java.util.*;
 
 
public class neliot {
 
 
public static int find(int[] a, int target)
{
	for (int i = 0; i < a.length; i++)
		if (a[i] == target)
			return i;
 
	return -1;
}
 
 
	public static void main(String[] args) {
 
		Scanner input = new Scanner(System.in);
 
		int t = input.nextInt();
		String[] ans = new String[t];
		int a = 0;
		int b = 0;
		int a2 = 0;
		int b2 = 0;
 
		for (int t1 = 0; t1 < t; t1++) {
 
			int n = input.nextInt();
 
			ans[t1] = "NO";
 
			if (n < 3) {
				ans[t1] = "YES";
				continue;
			}
 
		// n >= 3
 
			int maxVal1 = (int) Math.sqrt(n/2);
			int maxVal2 = (int) Math.sqrt(n);
 
			for (int i = maxVal2; i >= maxVal1; i--) {
				a = (int) Math.sqrt(n-b);
				a2 = a * a;
				b2 = i * i;
				if ((a2 + b2) == n) {
					ans[t1] = "YES";
					break;
				}
 
			}
 
		}  // t1
 
 
		for (int i = 0; i < t; i++) {
			System.out.println(ans[i]);
		}
 
		input.close(); 
    }
}

Test 1

Group: 1, 2

Verdict: WRONG ANSWER

Test 2

Group: 1, 2

Verdict: WRONG ANSWER

Test 3

Group: 2

Verdict: WRONG ANSWER