寻找二叉树指定节点的祖先节点

目前有两种方法：

1. 通过后序非递归遍历来寻找指定节点的所有祖先。代码示例如下：

// 该方法的得到的path中祖先顺序：根->自己 对应 [size-1, 0]
// 所以在寻找最近公共祖先时需要先进行反转操作。当然也可以在加入path时就行调整
void getAnstor(TreeNode* rt, vector<int> &path, int p){
        if(rt == nullptr) return ;
        stack<TreeNode*> stack_;
        unordered_set<TreeNode*> hash;// 标记节点是否访问过
        stack_.push(rt);
        TreeNode* temp = nullptr;
        while(!stack_.empty()){
            temp = stack_.top();
            if((temp->left == nullptr || hash.count(temp->left) != 0) && 
               (temp->right == nullptr || hash.count(temp->right) != 0)){// 左右节点为空或都已经被访问过，栈顶元素为自己，栈中除自己外的元素都为自己的祖先节点
                if(temp->val == p){
                    while(!stack_.empty()){
                        path.push_back(stack_.top()->val);
                        stack_.pop();
                    }
                    break;
                }else{
                    stack_.pop();
                }
            }else if(temp->left && hash.count(temp->left) == 0){ // 如果左节点不为空且没有被访问
                hash.insert(temp->left);
                stack_.push(temp->left);
            }else if(temp->right && hash.count(temp->right) == 0){ // 如果右节点不为空且没有被访问
                hash.insert(temp->right);
                stack_.push(temp->right);
            }
        }
    }

2. 利用先序遍历。该方法用于寻找指定节点的祖先节点，常用于最近公共祖先题型

bool flag = true;
void helper(TreeNode* rt, vector<int> &path, int o){
        if(!flag || !rt) return;
        // 先序
        path.push_back(rt->val);
        if(rt->val == o){
            flag = false;
            return;
        }
        helper(rt->left, path, o);
        helper(rt->right, path, o);
        if(flag){// 在上述函数返回时，flag标记没有改变，表示该子树没有对应值，应该将该子树弹出
            path.pop_back();
        }
    }

BM38 在二叉树中找到两个节点的最近公共祖先中有两种方法对应的代码，供参考学习。

class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        if (root == q || root == p || root == NULL) return root;
        TreeNode* left = lowestCommonAncestor(root->left, p, q);
        TreeNode* right = lowestCommonAncestor(root->right, p, q);
        if (left != NULL && right != NULL) return root;
        if (left == NULL) return right;
        return left;
    }
};