107、【树与二叉树】leetcode ——236. 二叉树的最近公共祖先:回溯法(C++版本)

题目描述

107、【树与二叉树】leetcode ——236. 二叉树的最近公共祖先:回溯法(C++版本)_第1张图片
107、【树与二叉树】leetcode ——236. 二叉树的最近公共祖先:回溯法(C++版本)_第2张图片
107、【树与二叉树】leetcode ——236. 二叉树的最近公共祖先:回溯法(C++版本)_第3张图片
107、【树与二叉树】leetcode ——236. 二叉树的最近公共祖先:回溯法(C++版本)_第4张图片
236. 二叉树的最近公共祖先

解题思路

p和q可能会出现两种情况:
(1)p和q分别在左、右子树中,此时返回最近公共祖先结点;
(2)p和q在同一侧子树中,此时谁为祖先结点返回谁。
107、【树与二叉树】leetcode ——236. 二叉树的最近公共祖先:回溯法(C++版本)_第5张图片
107、【树与二叉树】leetcode ——236. 二叉树的最近公共祖先:回溯法(C++版本)_第6张图片

回溯法

自底向上的返回信息,采用后序遍历的方式。当从左孩子或右孩子中找到p或q,则返回这个结点。这个结点再想上传递给上面的结点。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        // 遍历到空节点返回NULL,找到q或者p返回root。注意当出现情况二时,先找到谁,一定会先返回谁。
        if(!root || root == p || root == q)           return root;
        // 左右
        TreeNode* left = lowestCommonAncestor(root->left, p, q);
        TreeNode* right = lowestCommonAncestor(root->right, p, q);
        // 中:当左和右都不为NULL,说明当前为公共祖先,返回root
        if(left && right)       return root;
        // 当一侧为零,另一侧不为零,返回孩子
        else if(!left)          return right;
        return left;
    }
};

107、【树与二叉树】leetcode ——236. 二叉树的最近公共祖先:回溯法(C++版本)_第7张图片

参考文章:
236. 二叉树的最近公共祖先

你可能感兴趣的:(数据结构与算法刷题,#,树与二叉树,#,回溯算法,leetcode,c++,算法)