[Basic] Power of 2

Given a positive integer N, check if N is a power of 2.

Input:
The first line contains ‘T’ denoting the number of test cases. Then follows description of test cases. Each test case contains a single positive integer N.

Output:
Print “YES” if it is a power of 2 else “NO”. (Without the double quotes)

Constraints:
1<=T<=100
0<=N<=10^18

Example:
Input :
2
1
98

Output :
YES
​NO

Explanation: (2^0 == 1)

def powerOf2(n):
    if n == 1:
        return print("YES")
    elif n == 0:
        return print("NO")
    else:
        while n // 2 != 0:
            if n % 2 != 0:
                return print("NO")
            else:
                n = n // 2
    return print("YES")





t = int(input())

for i in range(t):
    n = int(input())
    powerOf2(n)

Comments