leetcode.199 두 갈래 나무의 오른쪽 보기

2812 단어
두 갈래 나무를 정해서 오른쪽에 서 있는 것을 상상하고 꼭대기에서 끝까지 순서대로 오른쪽에서 볼 수 있는 노드 값을 되돌려줍니다.
예:
입력: [1,2,3,null, 5,null, 4] 출력: [1,3,4] 설명:
1 /\2 3 \\5 4
출처: 리코드(LeetCode) 링크:https://leetcode-cn.com/problems/binary-tree-right-side-view저작권은 인터넷 소유에 귀속된다.상업 전재는 정부에 연락하여 권한을 부여하고, 비상업 전재는 출처를 명시해 주십시오.
 
차원을 두루 활용하여 각 층의 마지막 점을 기록하고 너무 자세하게 생각하지 않고 마음대로 썼다.
/**
 * 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:
    vector<int> rightSideView(TreeNode* root) {
        TreeNode *p[10000],*q;
        vector<int> res;
        if(root==NULL){
            return res;
        }
        int rear=0,front=0,level=0,pos=0;
        p[rear++]=root;
        while(rear!=front){
            q=p[front++];
            if(q->left!=NULL){
                p[rear++]=q->left;
            }
            if(q->right!=NULL){
                p[rear++]=q->right;
            }
            if(front-1==pos){
                res.push_back(p[pos]->val);
                pos=rear-1;
                level++;
            }
        }
        return res;
    }
};

좋은 웹페이지 즐겨찾기