螺旋折线python

螺旋折线
如图所示的螺旋折线经过平面上所有整点恰好一次。

对于整点(X, Y)，我们定义它到原点的距离dis(X, Y)是从原点到(X, Y)的螺旋折线段的长度。
例如dis(0, 1)=3, dis(-2, -1)=9
给出整点坐标(X, Y)，你能计算出dis(X, Y)吗？
　　
输入格式

X和Y

对于40%的数据，-1000 <= X, Y <= 1000 　　
对于70%的数据，-100000 <= X， Y <= 100000
对于100%的数据, -1000000000 <= X, Y <= 1000000000

输出格式

输出dis(X, Y)

样例输入

0 1
1
样例输出

3

X,Y = map(int,input().split())

n = max(abs(X),abs(Y)) #第几个正方形
if X < 0 and  X == Y: #特殊情况
    print(4*n*(n+1))
else:
    out = 4*n*(n-1) #前n-1个正方形的周长和
	#(X,Y)与(-n,-n)的距离
    if X == -n:
        d = Y + n
    elif Y == n:
        d = X + 3*n
    elif X == n:
        d = 5*n - Y
    else:
        d = 7*n - X
    print(out + d)

为什么
if X == -n:
d = Y + n这部分不用再管y的值，因为加的y负值的时候会抵消。