層次遍歷

我的想法：層序遍歷，然後輸出每一層最右邊的節點程式碼如下：

class Solution {
public:
    vector<int> rightSideView(TreeNode* root) {
        vector<int> result;
        vector<TreeNode*> high;
        vector<TreeNode*> low;
        if(root==NULL) return result;
        high.push_back(root);
        while(high.size()!=0){
            result.push_back(high[high.size()-1]->val);
            for(int i=0;i<high.size();i++){
                if(high[i]->left!=NULL) low.push_back(high[i]->left);
                if(high[i]->right!=NULL) low.push_back(high[i]->right);
            }
            high=low;
            low.clear();
        }
        return result;
    }
};
 

參考討論區的回答後，採用了改進的先序遍歷來做，遍歷順序為根-》右子樹-》左子樹。當一個節點被遍歷到的時候，如果結果集中的節點數比當前的level值小的話，說明當前節點是本層中第一個被遍歷到的，因為我們先遍歷右子樹，所以當前節點一定是本層中最右節點。程式碼如下

class Solution {
public:
    void preOrder(TreeNode* root,int level,vector<int>&res){
        if(root==NULL) return;
        if(level>res.size()) res.push_back(root->val);
        preOrder(root->right,level+1,res);
        preOrder(root->left,level+1,res);
    }
    vector<int> rightSideView(TreeNode* root) {
        vector<int> res;
        preOrder(root,1,res);
        return res;
    }
};