LeetCode | 두 갈래 나무의 차원 훑어보기 II

1433 단어 LeetcodeLeetCode
두 갈래 나무를 정해서 노드 값이 밑에서 위로 올라가는 차원을 되돌려줍니다.(즉, 잎 노드가 있는 층에서 뿌리 노드가 있는 층으로 한 층씩 왼쪽에서 오른쪽으로 옮겨간다)
예:
  [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7

아래에서 위로 올라가는 단계를 다음과 같이 되돌려줍니다.
[
  [15,7],
  [9,20],
  [3]
]

문제 풀이 사고방식이 두루 흐르고 노드를 체인 테이블에 넣고 저장하며 사용할 때마다 팝이 나온다.
/**
 * 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> levelOrderBottom(TreeNode* root) {
        vector> ret;
        
        if(NULL == root)
            return ret;

        queue q;
        q.push(root);
        
        while(!q.empty())
        {
            vector layer_vec;
            int lays = q.size();

            while(lays > 0) //  
            {
                TreeNode* tmp = q.front();
                q.pop();
                
                layer_vec.push_back(tmp->val);
                
                if(NULL != tmp->left)
                    q.push(tmp->left);
                if(NULL != tmp->right)
                    q.push(tmp->right);
                lays--;
            }
            ret.insert(ret.begin(), layer_vec);
        }
        return ret;
    }
};

좋은 웹페이지 즐겨찾기