python - 如何编写 __iter__ 从叶节点迭代回根？

Question

我正在实现搜索算法 (BFS) 并具有以下 Node 类:

class Node:
    def __init__(self, state=None, action=None, path_cost=None, parent=None):
        self._state = state
        self._action = action
        self._path_cost = path_cost
        self._parent = parent

我的 BFS 求解器返回解节点(子节点)。例如，使用此节点，我可以计算总路径成本，如下所示(此代码是另一个 Summary 类的一部分):

    def path_cost(self):
        self._cost = self._node.path_cost
        node = self._node.parent
        while node:
            self._cost += node.path_cost
            node = node.parent

        return self._cost

是否有更好的方法通过在 Node 中创建自定义 __iter__ 方法来实现此目的？

Answer 1

像这样的生成器函数可以工作:

class Node:
    def __iter__(self):
        node = self
        while node:
            yield node
            node = node._parent

# elsewhere
cost = sum(n.path_cost for n in self._node)
    # ....