1515B - Phoenix and Puzzle.cpp

/*
Phoenix is playing with a new puzzle, which consists of n identical puzzle pieces. Each puzzle piece is a right isosceles triangle as shown below.

A puzzle piece
The goal of the puzzle is to create a square using the n pieces. He is allowed to rotate and move the pieces around, but none of them can overlap and all n pieces must be used (of course, the square shouldn't contain any holes as well). Can he do it?

Input
The input consists of multiple test cases. The first line contains an integer t (1=t=104) — the number of test cases.

The first line of each test case contains an integer n (1=n=109) — the number of puzzle pieces.

Output
For each test case, if Phoenix can create a square with the n puzzle pieces, print YES. Otherwise, print NO.

Example
inputCopy
3
2
4
6
outputCopy
YES
YES
NO
*/



#include<bits/stdc++.h>
using namespace std;
 
int main(){
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    
    int t;
    cin >> t;
    
    while (t--) {
    	int n;
    	cin >> n;
    	
    	int a = sqrt(n / 2);
    	int b = sqrt(n / 4);
    	
    	a * a * 2 == n || b * b * 4 == n ? cout << "YES\n" : cout << "NO\n";
    	
	}
  	
    return 0;
}