x = int(input("请输入x的值"))

if x > 0 :

    y = sqrt(2 * x) + 1

elif x == 0:

    y = 0

else:

    y = x ** 2 -1

print (y)