Create smart_num.py

Author: puncoder

smart_num.py

@@ -0,0 +1,17 @@
+# https://www.hackerrank.com/challenges/smart-number/problem
+
+import math
+
+def is_smart_number(num):
+    val = int(math.sqrt(num))
+    if num / (val*val) == 1:
+        return True
+    return False
+
+for _ in range(int(input())):
+    num = int(input())
+    ans = is_smart_number(num)
+    if ans:
+        print("YES")
+    else:
+        print("NO")