计算一个给定范围的未知数的值

Question

输入一个正整数n,计算出大于n并且加上168后是一个完全平方数的最小数并输出
例:
输入 1
输出 28

Answer 1

num=int(input())
count=0
for i in range(num+1,1000):
    for j in range(1,1000):
        if (i+168)==j*j:
            print(i)
            count=1
            break
    if count==1:
        break