求助：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/

1. class Solution:
   
2.     def distanceK(self, root: TreeNode, target: TreeNode, K: int) -> List[int]:
   
3.         res = []
   
4.         seen = []
   
5.         self.attachParent(root)
   
6.         self.dfs(target, K, seen, res)
   
7.         return res
   
8. 
   
9.     def attachParent(self, node, par = None):
   
10.         if node:
    
11.             node.par = par
    
12.             self.attachParent(node.left, node)
    
13.             self.attachParent(node.right, node)
    
14.         
    
15.     def dfs(self, node, k, seen, res):
    
16.         if not node or node in seen:
    
17.             return
    
18.         seen.append(node.val)
    
19.         if k == 0:
    
20.             res.append(node.val)
    
21.             return
    
22.         self.dfs(node.left, k-1, seen, res)
    
23.         self.dfs(node.right, k-1, seen, res)
    
24.         self.dfs(node.par, k-1, seen, res)

复制代码

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

1. class Solution:
   
2.     def distanceK(self, root: TreeNode, target: TreeNode, K: int) -> List[int]:
   
3.         res = []
   
4.         seen = []
   
5.         self.attachParent(root)
   
6.         self.dfs(target, K, seen, res)
   
7.         return res
   
8. 
   
9.     def attachParent(self, node, par = None):
   
10.         if node:
    
11.             node.par = par
    
12.             self.attachParent(node.left, node)
    
13.             self.attachParent(node.right, node)
    
14.          
    
15.     def dfs(self, node, k, seen, res):
    
16.         if not node or node.val in seen:
    
17.             return
    
18.         seen.append(node.val)
    
19.         if k == 0:
    
20.             res.append(node.val)
    
21.             seen.append(node.val)
    
22.             return
    
23.         self.dfs(node.left, k-1, seen, res)
    
24.         self.dfs(node.right, k-1, seen, res)
    
25.         self.dfs(node.par, k-1, seen, res)

复制代码

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

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