993. 二叉树的堂兄弟节点

Posted 2023-03-07 Debroon

993. 二叉树的堂兄弟节点

• • 题目
  • 算法设计：深度优先搜索

 

---

题目

传送门：https://leetcode.cn/problems/cousins-in-binary-tree/

 

---

算法设计：深度优先搜索

遍历找到 x，y 的深度和父节点，对比即可。

class Solution 
public:
    bool isCousins(TreeNode* root, int x, int y) 
		unordered_map<int, pair<int, int>> m;
		dfs(root, 0, -1, m);
		return m[x].first == m[y].first && m[x].second != m[y].second;
		// 判断 x，y 是否是表兄弟节点
    
    void dfs(TreeNode* root, int depth, int parent, unordered_map<int, pair<int, int>>& m) 
    	if (!root) return;
    	m[root->val] = depth, parent;            // 记录深度和父节点
    	dfs(root->left, depth + 1, root->val, m);
    	dfs(root->right, depth + 1, root->val, m);
    
;