全局变量重置循环？

Question

我试图从代码问世开始解决问题，但似乎无法破解。 它应该收集一组唯一的坐标，然后输出其计数。 我的主要变量coords_list在for循环外部实例化，然后添加到for内部。

但是由于某种原因，每次我在for附加它时，它似乎都“复位”，所以最大程度上，我的返回列表是coords_list初始值列表以及我附加到它的最新值的列表。

我很可能只是缺少一些简单的东西。 尽管如此，任何帮助还是很棒的。

输出量

coords_list: [[0, 0]]
START -- coords: [0, 0], vector: (1, 0), instruction: >
END -- coords: [1, 0], coords_list: [[0, 0], [1, 0]]
input: >, len: 2
------
coords_list: [[0, 0]]
START -- coords: [0, 0], vector: (0, 1), instruction: ^
END -- coords: [0, 1], coords_list: [[0, 0], [0, 1]]
START -- coords: [0, 1], vector: (1, 0), instruction: >
END -- coords: [1, 1], coords_list: [[0, 0], [1, 1]]
START -- coords: [1, 1], vector: (0, -1), instruction: v
END -- coords: [1, 0], coords_list: [[0, 0], [1, 0]]
START -- coords: [1, 0], vector: (-1, 0), instruction: <
END -- coords: [0, 0], coords_list: [[0, 0], [0, 0]]
input: ^>v<, len: 2
------
Traceback (most recent call last):
  File "python", line 47, in <module>
AssertionError

码

def return_vector(chary):

    if chary == "^":
        ret_value = (0,1)
    elif chary == "v":
        ret_value = (0,-1)
    elif chary == ">":
        ret_value = (1, 0)
    elif chary == "<":
        ret_value = (-1, 0)
    else:
        raise ValueError("Expecting one of '^','v','>','<', got {}".format(chary))

    return ret_value

def main(input):
    coords_list = [[0,0]]
    coords = [0,0]
    lst_input = list(input)
    debug_iterator, debug_loop = 0, 10
    print("coords_list: {}".format(coords_list))

    for instruction in lst_input:
        vector = return_vector(instruction)


        if debug_iterator < debug_loop:
            print("START -- coords: {}, vector: {}, instruction: {}".format(coords, vector, instruction))

        coords[0] += vector[0]
        coords[1] += vector[1]

        if coords not in coords_list:
            coords_list.append(coords)

        if debug_iterator < debug_loop:
            print("END -- coords: {}, coords_list: {}".format(coords, coords_list))
            debug_iterator += 1

    print('input: {}, len: {}\n------'.format(input, len(coords_list)))
    return(len(coords))

assert main('>') == 2
assert main('^>v<') == 4
assert main('^v^v^v^v^v') == 2

print("main(INPUT) == {}".format(main(INPUT)))


assert return_vector('^') == (0,1)

Answer 1

编写coords_list.append(coords) ，您将对一个coords对象附加一个引用，然后对其进行更改。 相反，您需要一个副本。 您可以使用coords[:]来做到这一点，但是在这里使用不可变元组或namedtuple更合适。 怎么样

def main(inp):
    coords = set()
    cur = (0, 0)
    coords.add(cur)
    for instruction in inp:
        vector = return_vector(instruction)
        cur = (cur[0] + vector[0], cur[1] + vector[1])
        coords.add(cur)
    return len(coords)

请注意其他一些更改：

• coords_list实际上更多。 重命名为coords并进行设置。 这简化了添加，并使大型坐标集的成员资格测试更快。
• 重命名了input ，因为它是内置的
• list(input)已经遍历了输入。 然后，您遍历该列表。 取而代之的是，遍历输入一次更快，更容易。
• 删除调试代码
• 使用两个变量作为当前位置。 使用元组作为当前位置。