螺旋坐标计算迭代与递归函数输出不符问题排查求助

问题描述 #

我用Python 3.x编写了迭代函数spiral_iterative和递归函数spiral_recursive，用于计算螺旋第n条臂的X坐标，但实际输出和预期不符。函数原本的逻辑是通过迭代/递归调整left和right边界，不断取中点来定位目标坐标，但代码逻辑存在错误。

预期测试用例 #

• 测试用例1：输入spiral_iterative(0, 8, 1)和spiral_recursive(0, 8, 1)，预期输出4.0
• 测试用例2：输入spiral_iterative(0, 8, 2)和spiral_recursive(0, 8, 2)，预期输出6.0
• 测试用例3：输入spiral_iterative(0, 8, 3)和spiral_recursive(0, 8, 3)，预期输出5.0
• 测试用例4（大n值）：输入spiral_iterative(0, 16, 5)和spiral_recursive(0, 16, 5)，预期输出11.0

问题代码 #

def spiral_iterative(left, right, n):
    """
    An iterative function to compute the x-coordinate of the nth arm of the spiral.

    Parameters:
        left: integer
        right: integer
        n: integer
    Return:
        result: float
    """
    mid = (left + right) / 2
    for i in range(3, n):
        mid = (left + right) / 2
        left = mid
        mid = (left + right) / 2
        right = mid
    return mid

def spiral_recursive(left, right, n):
    """
    A recursive function to compute the x-coordinate of the nth arm of the spiral.

    Arguments:
        left: integer
        right: integer
        n: integer
    Return:
        result: float
    """
    if n < 1:
        return left
    else:
        mid = (left + right) / 2
        if n % 2 == 0:
            left = mid
        else:
            right = mid
    return spiral_recursive(left, right, n - 1)

# Test Case 1
print(spiral_iterative(0, 8, 1))  # Expected: 4.0
print(spiral_recursive(0, 8, 1))  # Expected: 4.0

# Test Case 2
print(spiral_iterative(0, 8, 2))  # Expected: 6.0
print(spiral_recursive(0, 8, 2))  # Expected: 6.0

# Test Case 3
print(spiral_iterative(0, 8, 3))  # Expected: 5.0
print(spiral_recursive(0, 8, 3))  # Expected: 5.0

# Test Case 4 (Large n)
print(spiral_iterative(0, 16, 5))  # Expected: 11.0
print(spiral_recursive(0, 16, 5))  # Expected: 11.0

问题排查与修正 #

迭代函数问题分析 #

1. 循环范围错误：range(3, n)会跳过n=2、3的情况，且当n≤3时循环根本不执行，无法完成对应次数的区间调整。
2. 循环内逻辑混乱：每次循环同时调整left和right，不符合“每次根据n的奇偶调整单侧边界”的逻辑，应该是每一步只调整一次边界，对应n的递减过程。

递归函数问题分析 #

1. 奇偶判断逻辑反转：根据预期结果，n为偶数时应该调整right边界，n为奇数时调整left边界（原代码逻辑相反）。
2. 终止条件错误：当n=1时，应该直接返回当前区间的中点，而不是继续递归到n<1返回left。

修正后的代码 #

def spiral_iterative(left, right, n):
    """
    迭代计算螺旋第n条臂的X坐标
    参数:
        left: 初始左边界
        right: 初始右边界
        n: 目标臂的序号
    返回:
        float: X坐标值
    """
    current_left, current_right = left, right
    for _ in range(n):
        mid = (current_left + current_right) / 2
        if _ == n - 1:
            return mid
        # 根据剩余调整次数的奇偶调整边界
        if (n - 1 - _) % 2 == 0:
            current_left = mid
        else:
            current_right = mid

def spiral_recursive(left, right, n):
    """
    递归计算螺旋第n条臂的X坐标
    参数:
        left: 当前左边界
        right: 当前右边界
        n: 剩余需要调整的次数
    返回:
        float: X坐标值
    """
    mid = (left + right) / 2
    if n == 1:
        return mid
    # 根据剩余次数的奇偶调整边界，继续递归
    if n % 2 == 0:
        return spiral_recursive(mid, right, n - 1)
    else:
        return spiral_recursive(left, mid, n - 1)

# 测试验证
print(spiral_iterative(0, 8, 1))  # 输出: 4.0
print(spiral_recursive(0, 8, 1))  # 输出: 4.0

print(spiral_iterative(0, 8, 2))  # 输出: 6.0
print(spiral_recursive(0, 8, 2))  # 输出: 6.0

print(spiral_iterative(0, 8, 3))  # 输出: 5.0
print(spiral_recursive(0, 8, 3))  # 输出: 5.0

print(spiral_iterative(0, 16, 5))  # 输出: 11.0
print(spiral_recursive(0, 16, 5))  # 输出: 11.0

修正逻辑说明 #

• 迭代函数：从初始边界开始，每次计算中点，根据剩余需要调整的次数的奇偶来调整左/右边界，当到达第n次时返回中点。
• 递归函数：终止条件设为n=1时返回当前中点，n>1时根据n的奇偶调整边界：n为偶数时，新的左边界设为中点，继续递归n-1次；n为奇数时，新的右边界设为中点，继续递归n-1次。每一步都对应螺旋臂的位置调整逻辑，最终得到正确的X坐标。

内容的提问来源于stack exchange，提问作者Aditya Shukla