蓝桥杯 螺旋折线

Code

x, y = map(int, input().split(' '))

# 累加公式乘2
def preSum(n): 
    return n ** 2 + n

if x > 0:

    if abs(x) >= abs(y):
        ans = preSum(2 * x - 1) + 2 * x
        if y <= 0: ans += x + abs(y)  #
        else: ans += x - y  # 
    elif abs(x) < abs(y):
        if y <= 0: ans = preSum(2 * abs(y)) + abs(y) - abs(x)  #
        else: ans = preSum(2 * abs(y) - 1) + abs(y) + abs(x) #

elif x < 0:
    if abs(x) >= abs(y):
        ans = preSum(2 * abs(x) - 2) + 2 * abs(x) - 1
        if y < 0: ans += abs(x) -1 - abs(y) # 
        else: ans += (abs(x) - 1) + y # 
    elif abs(x) < abs(y):
        if y < 0: ans = preSum(2 * abs(y)) + abs(y) + abs(x) #
        else: ans = preSum(2 * abs(y) - 1) + abs(y) - abs(x) #

else:
    if y >= 0: ans = preSum(2 * y - 1) + y
    else: ans = preSum(2 * y - 1) - y

print(ans)