求助：863. All Nodes Distance K in Binary Tree 代码哪里出了问题呢

能不能请地里哪位大神帮忙看一下，下面的这段代码是哪里出了问题呢？为什么跑出来是wrong answer呢？
太感谢啦！
PS. LC的discussion改版之后就看不到我自己发的求助贴了，因为把newest这些sorted by都拿掉了。。心累┓(;′_｀)┏
题目在这里：https://leetcode.com/problems/al ... e-k-in-binary-tree/

[Python] 纯文本查看 复制代码

class Solution:
    def distanceK(self, root: TreeNode, target: TreeNode, K: int) -> List[int]:
        res = []
        seen = []
        self.attachParent(root)
        self.dfs(target, K, seen, res)
        return res

    def attachParent(self, node, par = None):
        if node:
            node.par = par
            self.attachParent(node.left, node)
            self.attachParent(node.right, node)
        
    def dfs(self, node, k, seen, res):
        if not node or node in seen:
            return
        seen.append(node.val)
        if k == 0:
            res.append(node.val)
            return
        self.dfs(node.left, k-1, seen, res)
        self.dfs(node.right, k-1, seen, res)
        self.dfs(node.par, k-1, seen, res)

帮你改了一下，应该就是你存入或者检查seen的时候要存或者check val不是整个node. 你在check的时候应该是笔误少打了个.val

[Python] 纯文本查看 复制代码

class Solution:
    def distanceK(self, root: TreeNode, target: TreeNode, K: int) -> List[int]:
        res = []
        seen = []
        self.attachParent(root)
        self.dfs(target, K, seen, res)
        return res
 
    def attachParent(self, node, par = None):
        if node:
            node.par = par
            self.attachParent(node.left, node)
            self.attachParent(node.right, node)
         
    def dfs(self, node, k, seen, res):
        if not node or node.val in seen:
            return
        seen.append(node.val)
        if k == 0:
            res.append(node.val)
            seen.append(node.val)
            return
        self.dfs(node.left, k-1, seen, res)
        self.dfs(node.right, k-1, seen, res)
        self.dfs(node.par, k-1, seen, res)

luke77gu 发表于 2020-11-24 02:37
帮你改了一下，应该就是你存入或者检查seen的时候要存或者check val不是整个node. 你在check的时候应该是笔 ...

啊。。。。原来是这样，还是大神眼尖而且有经验！
太感谢了呀！！！