20190702-剑指offer68-AcWing-88. 树中两个结点的最低公共祖先

剑指offer68-AcWing-88. 树中两个结点的最低公共祖先

给出一个二叉树，输入两个树节点，求它们的最低公共祖先。

一个树节点的祖先节点包括它本身。

注意：
输入的二叉树不为空；
输入的两个节点一定不为空，且是二叉树中的节点；

样例

二叉树[8, 12, 2, null, null, 6, 4, null, null, null, null]如下图所示：
    8
   / \
  12  2
     / \
    6   4

1. 如果输入的树节点为2和12，则输出的最低公共祖先为树节点8。
2. 如果输入的树节点为2和6，则输出的最低公共祖先为树节点2。

思路：

     1.先找到两个结点的路径，并保存在vector中 2.在vector中找到其最低公共祖先的结点。

C++ code:

/**
 * 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:
    void FindPath(TreeNode* root, TreeNode *target, int flag, vector<TreeNode*>& path, vector<TreeNode*>& result){
        if(root == NULL || flag == 1){
            return;
        }
        path.push_back(root);
        if(root->val == target->val){
            flag = 1;
            result = path;
            return;
        }
        FindPath(root->left, target, flag, path, result);
        FindPath(root->right, target, flag, path, result);
        path.pop_back();
    }
    
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        if(root == NULL || p == NULL || q == NULL){
            return NULL;
        }
        int flag = 0;
        vector<TreeNode*> path;
        vector<TreeNode*> p_path;
        vector<TreeNode*> q_path;
        FindPath(root, p, flag, path, p_path);
        path.clear();
        flag = 0;
        FindPath(root, q, flag, path, q_path);
        
        TreeNode* res = NULL;
        for(int i = 0; i < p_path.size(); i++){
            if(p_path[i] == q_path[i]){
                res = p_path[i];
            }else{
                break;
            }
        }
        return res;
    }
};