刷了几道BFS在二叉树方面运用的超级简单题,大致有了个模板,先看下今天的简单题
在二叉树中,根节点位于深度 0 处,每个深度为 k 的节点的子节点位于深度 k+1 处。
如果二叉树的两个节点深度相同,但父节点不同,则它们是一对堂兄弟节点。
我们给出了具有唯一值的二叉树的根节点 root,以及树中两个不同节点的值 x 和 y。
只有与值 x 和 y 对应的节点是堂兄弟节点时,才返回 true。否则,返回 false。
示例 1:
输入:root = [1,2,3,4], x = 4, y = 3 输出:false
示例 2:
输入:root = [1,2,3,null,4,null,5], x = 5, y = 4 输出:true
示例 3:
输入:root = [1,2,3,null,4], x = 2, y = 3 输出:false 提示:二叉树的节点数介于 2 到 100 之间。 每个节点的值都是唯一的、范围为 1 到 100 的整数。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/cousins-in-binary-tree
一看就是可以直接套用前面写二叉树的模板题,就其实都是从根遍历二叉树,先看下代码,与之前的不同的是多了父节点以及节点值的记录
/**
* 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 {
public:
bool isCousins(TreeNode* root, int x, int y) {
int fx=NULL,fy=NULL;
int dpx,dpy;
int depth=0;
queue<TreeNode*>q;
if(root!=NULL)
q.push(root);
while(!q.empty())
{
depth++;
for(int i=q.size()-1;i>=0;i--)
{
TreeNode* temp=q.front();
q.pop();
if(temp->left!=NULL)
{
q.push(temp->left);
if(temp->left->val==x)
{
fx=temp->val;
dpx=depth;
}
if(temp->left->val==y)
{
fy=temp->val;
dpy=depth;
}
}
if(temp->right!=NULL)
{
q.push(temp->right);
if(temp->right->val==x)
{
fx=temp->val;
dpx=depth;
}
if(temp->right->val==y)
{
fy=temp->val;
dpy=depth;
}
}
if(fx!=NULL&&fy!=NULL)
break;
}
}
if( fx!= fy && dpx==dpy)
return true;
else return false;
}
};
大致上的模板就是和下面求深度的方法一样就是一些细节方面要具体 问题具体分析。
int depth(TreeNode* root)
{
int depth=0;
queue<TreeNode*>q;
if(root!=NULL)
q.push(root);
while(!q.empty())
{
depth++;
for(int i=q.size()-1;i>=0;i--)
{
TreeNode* temp=q.front();
q.pop();
if(temp->left!=NULL)
{
q.push(temp->left);
}
if(temp->right!=NULL)
{
q.push(temp->right);
}
}
}
return depth;
}
这是我一写就A,完全没看题解区的为数不多的一题之一,我又快乐了!
菜鸟的快乐就是这么朴实无华。