生活随笔
收集整理的這篇文章主要介紹了
leetcode——第993题——二叉树的堂兄弟节点
小編覺(jué)得挺不錯(cuò)的,現(xiàn)在分享給大家,幫大家做個(gè)參考.
題目:
在二叉樹(shù)中,根節(jié)點(diǎn)位于深度 0 處,每個(gè)深度為 k 的節(jié)點(diǎn)的子節(jié)點(diǎn)位于深度 k+1 處。
如果二叉樹(shù)的兩個(gè)節(jié)點(diǎn)深度相同,但 父節(jié)點(diǎn)不同 ,則它們是一對(duì)堂兄弟節(jié)點(diǎn)。
我們給出了具有唯一值的二叉樹(shù)的根節(jié)點(diǎn) root ,以及樹(shù)中兩個(gè)不同節(jié)點(diǎn)的值 x 和 y 。
只有與值 x 和 y 對(duì)應(yīng)的節(jié)點(diǎn)是堂兄弟節(jié)點(diǎn)時(shí),才返回 true 。否則,返回 false。
/*** 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 {
/*
本題使用層序遍歷,也就是廣度優(yōu)先遍歷,也即是BST是最佳方案
但是我第一次在自己考慮的時(shí)候,僅僅考慮到需要再同一層,沒(méi)有考慮到不能是同一個(gè)父親
因此 看了題解,發(fā)現(xiàn)同一個(gè)父親的標(biāo)記也很簡(jiǎn)單
只需要在隊(duì)列中插入元素的時(shí)候,將父節(jié)點(diǎn)也插入,然后就可以對(duì)應(yīng)的判斷啦~~注意pair元素的賦值需要用 大括號(hào)
*/public:bool isCousins(TreeNode* root, int x, int y) {// 方法一:隊(duì)列里存放兩個(gè)樹(shù)節(jié)點(diǎn),一個(gè)是 cur,一個(gè)是 parent// 注意這里是que里是 pair 類(lèi)型,這樣的類(lèi)型,賦值操作都要用大括號(hào) queue<pair<TreeNode*, TreeNode*>> que;if(root == nullptr) return false;// 順序不要反了,當(dāng)前節(jié)點(diǎn) + 父節(jié)點(diǎn)que.push({root, nullptr});while(!que.empty()){int size = que.size(); // 更新每一層的 sizevector<TreeNode*> pare; // 記錄兩次的父節(jié)點(diǎn)for(int i = 0; i < size; ++i){auto[cur, parent] = que.front();que.pop(); // 這里不要忘記 pop尾刪~~~if(cur->val == x || cur->val == y) pare.push_back(parent);// 既要在同一層,又要不是同一個(gè)父親if(pare.size() == 2 && pare[0] != pare[1]) return true;if(cur->left != nullptr) que.push({cur->left, cur});//push的順序不要反了if(cur->right != nullptr) que.push({cur->right, cur});}if(pare.size() == 1) return false;}return false;}
};
總結(jié)
以上是生活随笔為你收集整理的leetcode——第993题——二叉树的堂兄弟节点的全部?jī)?nèi)容,希望文章能夠幫你解決所遇到的問(wèn)題。
如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò),歡迎將生活随笔推薦給好友。