LeetCode545.Boundary-of-Binary-Tree

1、头节点为边界节点
2、叶结点为边界节点
3、如果节点在其所在的层中是最左边或最右边，那么也是边界节点。

思路：分成三部分来做：找左边界结点、叶结点、右边界结点。

找左边界结点要遍历root的左子树，如果左孩子存在就加入vector，否则加入右孩子； 找叶结点，可以利用前序遍历，遍历结点改为判断结点是否是叶结点，是则加入；找右边界结点类似于找左边界结点，不过是其逆序，可以利用一个栈来辅助。 还要注意这三部分会有结点重合，在组合到一起的时候可以利用一个set来去掉重复的结点。注意不能在每个函数中用vector来返回结点中的值，否则无法去除重复的结点，因为树中结点的值不是唯一的，那就指针咯

#include <iostream>
#include <vector>
#include <stack>
#include <set>
using namespace std;

struct TreeNode {
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};

class Solution {
public:
    void leftMost(TreeNode* root, vector<TreeNode *> &vec)
    {
        while(root)
        {
            vec.push_back(root);
            if(root->left)
                root = root->left;
            else root = root->right;
        }
    }

    void leaf(TreeNode* root, vector<TreeNode *> &vec)
    {
        if(root)
        {
            if(!root->left && !root->right) vec.push_back(root);
            if(root->left)
                leaf(root->left, vec);
            if(root->right)
                leaf(root->right, vec);
        }
    }

    void rightMost(TreeNode* root, vector<TreeNode *> &vec)
    {
        stack<TreeNode *> st;
        while(root)
        {
            st.push(root);
            if(root->right)
                root = root->right;
            else root = root->left;
        }
        while(!st.empty())
        {
            vec.push_back(st.top());
            st.pop();
        }
    }

    vector<int> boundaryOfBinaryTree(TreeNode* root) {
        vector<int> ans;
        if(!root) return ans;
        vector<TreeNode *> tmp;
        set<TreeNode *> s;
        leftMost(root, tmp);
        leaf(root, tmp);
        rightMost(root, tmp);
        for(TreeNode *p : tmp)
        {
            if(s.find(p) == s.end())
            {
                ans.push_back(p->val);
                s.insert(p);
            }
        }
        return ans;
    }
};