MT1318 完美平方

题目

输入正整数N，检查它是否为完美平方。完美平方数是指1个平方数可以分成两部分后，每个部分仍然是平方数。如49=7 * 7，分成4和9，4和9都是平方数。再如1681=41*41，1681分成16和81，也都是平方数。

格式

输入格式

输入正整数N

输出格式

输出YES或者NO

样例1

输入：
49

输出：
YES

程序

def pingfangshu(x):
    # if x[0]=="0":
    #     return False
    x =int(x)
    n =int(x**(0.5))

    if n*n==x:
        return True
    else:
        return False
def main():
    #code here
    n = input()
    f =False
    # 1001->100,1 长度不等的两部分
    for i in range(1,len(n)):
        n1=n[:i]
        n2=n[i:]
        if pingfangshu(n) and pingfangshu(n1) and pingfangshu(n2):
            f = True
            break
    if f:
        print("YES")
    else:
        print("NO")

if __name__ == '__main__':
    main();