LintCode - 두 갈래 나무의 차원 반복 II

1922 단어 면접
두 갈래 나무를 제시하고 그 노드 값이 밑에서 위로 올라가는 차원으로 되돌아간다. (잎 노드가 있는 층에서 뿌리 노드가 있는 층으로 옮겨다닌 다음에 한 층씩 왼쪽에서 오른쪽으로 옮겨간다)
당신은 실제 면접에서 이 문제를 만난 적이 있습니까? 
Yes
예제
두 갈래 나무 한 그루를 주시오{3,9,20,#,#,15,7} ,
    3
   / \
  9  20
    /  \
   15   7

다음과 같이 아래에서 위로 이동합니다.
[
  [15,7],
  [9,20],
  [3]
]

태그
Expand  
분석: 층층이 두루 다니네...
코드:
/**
 * Definition of TreeNode:
 * class TreeNode {
 * public:
 *     int val;
 *     TreeNode *left, *right;
 *     TreeNode(int val) {
 *         this->val = val;
 *         this->left = this->right = NULL;
 *     }
 * }
 */
 
 
class Solution {
    /**
     * @param root : The root of binary tree.
     * @return : buttom-up level order a list of lists of integer
     */
public:
    vector> levelOrderBottom(TreeNode *root) {
        // write your code here
        vector > ret;
        if(root==nullptr)
            return vector >();
        vector cur;
        cur.push_back(root);
        ret.push_back(cur);
        while(true)
        {
            vector newCur;
            vector cur = ret.back();
            for(auto x:cur)
            {
                if(x->left)
                    newCur.push_back(x->left);
                if(x->right)
                    newCur.push_back(x->right);
            }
            if(newCur.size()>0)
                ret.push_back(newCur);
            else
                break;
        }
        reverse(ret.begin(),ret.end());
        vector > values;
        for(auto v:ret)
        {
            vector vals;
            for(auto t:v)
                vals.push_back(t->val);
            values.push_back(vals);
        }
        return values;
    }
};

좋은 웹페이지 즐겨찾기