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