검지offer - 두 갈래 나무를 여러 줄로 인쇄하기

2300 단어
화전북풍천진대학 인지계산 및 응용중점실험실 날짜: 2015/10/8
제목 설명은 위에서 아래로 층별로 두 갈래 트리를 인쇄하고 같은 층의 결점은 왼쪽에서 오른쪽으로 출력합니다.층마다 줄을 출력합니다.
해석: 이것은 줄에 따라 인쇄하는 것과 같은 사고방식으로 하면 된다.
/* struct TreeNode { int val; struct TreeNode *left; struct TreeNode *right; TreeNode(int x) : val(x), left(NULL), right(NULL) { } }; */
class Solution {
public:
    vector<vector<int> > Print(TreeNode* pRoot)
    {
        vector<vector<int>> result;
        if (pRoot == NULL)
            return result;
        vector<int> temp;
        TreeNode * flag = new TreeNode(NULL);
        queue<TreeNode *> q;
        q.push(pRoot);
        q.push(flag);
        while (true)
        {
            TreeNode * top = q.front();
            q.pop();
            if (top == flag)
            {
                result.push_back(temp);
                temp.clear();
                q.push(top);
                top = q.front();
                q.pop();
                if (top == flag)
                    break;
            }
            temp.push_back(top->val);
            if (top->left != NULL)
                q.push(top->left);
            if (top->right != NULL)
                q.push(top->right);
        }
        return result;
    }
};

좋은 웹페이지 즐겨찾기