a = int(input())
s = 1
x = 0
while s <= a:
s *= 2
x += 1
if s == a:
print('YES')
break
else:
print('NO')
Ещё такой вариант может быть, ближе твоему:
import math
b = math.log2(a)
if b%1 == 0: