标签:rt Distance right TreeNode Binary int Tree return left
Given the root
of a binary tree and two integers p
and q
, return the distance between the nodes of value p
and value q
in the tree.
The distance between two nodes is the number of edges on the path from one to the other.
Solution
求树上两个点之间的距离。很经典的做法就是求出 \(LCA\),然后分别求出两点分别到 \(LCA\) 的距离即可
点击查看代码
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
private:
TreeNode* LCA(TreeNode* rt, int p, int q){
if(!rt) return NULL;
if(rt->val==p || rt->val==q)return rt;
TreeNode* leftrt = LCA(rt->left, p,q);
TreeNode* rightrt = LCA(rt->right, p, q);
if(leftrt&&rightrt)return rt;
if(!leftrt && !rightrt)return NULL;
return leftrt?leftrt:rightrt;
}
int get_depth(TreeNode* rt, int v, int dep){
if(!rt) return 0;
if(rt->val==v) return dep;
return get_depth(rt->right, v, dep+1)+get_depth(rt->left, v, dep+1);
}
public:
int findDistance(TreeNode* root, int p, int q) {
TreeNode* lca = LCA(root, p, q);
return get_depth(lca,p,0)+get_depth(lca,q,0);
}
};
标签:rt,Distance,right,TreeNode,Binary,int,Tree,return,left 来源: https://www.cnblogs.com/xinyu04/p/16600690.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。