검지 offer - 지그재그 순서로 두 갈래 나무 인쇄

3445 단어
화전북풍천진대학 인지계산 및 응용중점실험실 날짜: 2015/10/8
제목 설명은 함수를 지그재그로 인쇄하는 두 갈래 트리, 즉 첫 번째 줄은 왼쪽에서 오른쪽으로, 두 번째 줄은 오른쪽에서 왼쪽으로, 세 번째 줄은 왼쪽에서 오른쪽으로, 다른 줄은 이와 같이 인쇄합니다.
해석: 앞에서 만났던 어떤 줄에 따라 두 갈래 나무를 인쇄하는 것과 유사하다. 차이점은 여기에 설정된 flag는 이동하지 않고 왼쪽에서 오른쪽으로 접근하는 것은 왼쪽에서 오른쪽으로, 오른쪽에서 왼쪽으로 접근할 때 오른쪽에서 왼쪽으로, 왼쪽에서 왼쪽으로 접근하면 된다.
/* 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;
        TreeNode * flag = new TreeNode(NULL);
        deque<TreeNode *> q;
        q.push_back(pRoot);
        q.push_back(flag);
        vector<int> temp;
        bool lefttoright = true;
        while (q.size()>1)
        {
            if (lefttoright)
            {
                TreeNode * top = q.front();
                if (top == flag)
                {
                    result.push_back(temp);
                    temp.clear();
                    lefttoright = false;
                    continue;
                }
                q.pop_front();
                temp.push_back(top->val);
                if (top->left != NULL)
                    q.push_back(top->left);
                if (top->right != NULL)
                    q.push_back(top->right);
            }
            else
            {
                TreeNode * end = q.back();
                if (end == flag)
                {
                    result.push_back(temp);
                    temp.clear();
                    lefttoright = true;
                    continue;
                }
                q.pop_back();
                temp.push_back(end->val);
                if (end->right != NULL)
                    q.push_front(end->right);
                if (end->left != NULL)
                    q.push_front(end->left);
            }
        }
        result.push_back(temp);
        return result;
    }
};

좋은 웹페이지 즐겨찾기