算法训练 大等于n的最小完全平方数 python

算法训练 大等于n的最小完全平方数

问题描述
　　输出大等于n的最小的完全平方数。
　　若一个数能表示成某个自然数的平方的形式，则称这个数为完全平方数
　　Tips：注意数据范围
输入格式
　　一个整数n
输出格式
　　大等于n的最小的完全平方数
样例输入
71711
样例输出
71824
数据规模和约定
　　n是32位有符号整数

我的代码：

n= int(input())
k=n//2
p=0
h=0
while True:
    if n<=0:
        print(0)
        break
    else:
        if k**2>=n:
            m=k
            p=1
        else:
            p=-1
            h=1

        if p==1 and h==0:
            k=k//2
        elif p==-1:
            k+=1

        if h==1 and p==1:
            print(m**2)
            break