leetcode--------All Nodes Distance K in Binary Tree

最新推荐文章于 2024-04-10 20:23:41 发布

抖抖认真了

最新推荐文章于 2024-04-10 20:23:41 发布

阅读量139

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签： leetcode

本文链接：https://blog.csdn.net/qq_41983709/article/details/90674089

leetcode 专栏收录该内容

100 篇文章

订阅专栏

博客围绕LeetCode题目展开，指出解题不能硬写，要利用结构。解题思路是用map记录父亲节点，仅记录target节点路径的父亲节点会使问题复杂。找节点时根据K的长度递归深度，需用容器记录节点是否已访问，还可对路径上的父亲节点按K深度递归。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

我写了一下，大体的逻辑是对的，但是要学会利用结构，而不是硬写，逻辑本来很简单，但也让你搞得很复杂！

这题得思路是，这就相当于一个单向的，不能访问父亲节点，所以我必须用一个map来记录所有的父亲节点，而我只是单纯的把只关于target的节点的一条路径的父亲节点给记录下来，这样也让我写的更加复杂，

而接下来的找的过程，也就是根据K的长度，递归K的深度，如果不为空，则record，关于父亲节点的递归，需要记录其这个节点是已访问的，就不能在访问了，就要递归它的另一个孩子，所以还需要一个容器来记录该节点是否已访问！！

一条路径上的父亲节点也可以这样根据K的深度进行递归！

下面的这个算法的描述是看了别人的解法思路写的。

class Solution {
public:
	

	std::vector<int> result;
	std::map< TreeNode*, TreeNode*> par_chil;
	std::set<TreeNode*> visitied;
	//利用层次遍历，其实也就是广度搜索，找到所有的父亲
	void find_parent(TreeNode* root)
	{
		if (!root)	return;
	
			if (root->left)
			{
				par_chil[root->left] = root;
				find_parent(root->left);
			}
			if (root->right)
			{	
				par_chil[root->right] = root;
				find_parent(root->right);
			}
				
			
			
		
		
	}

	void dfs(TreeNode* root, int curr_i)
	{
		if (!root)	return;	//递归基
		if (curr_i==0) {
			result.emplace_back(root->val);
			return;
		}
		visitied.insert(root);
		if (root->left&&(visitied.find(root->left)==visitied.end()))	dfs(root->left, curr_i - 1);
		if (root->right && (visitied.find(root->right) == visitied.end()))	dfs(root->right, curr_i - 1);
		TreeNode* parent = par_chil[root];
		if(parent&&visitied.find(parent) == visitied.end())	dfs(parent, curr_i - 1);
	}

	vector<int> distanceK(TreeNode* root, TreeNode* target, int K) {
		find_parent(root);
		dfs(target, K);
		return result;
	}
};