剑指offer——面试题61:按之字形顺序打印二叉树

来源:互联网 发布:人工智能的未来下载 编辑:程序博客网 时间:2024/05/16 14:46

    题目:请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推

    代码:

/*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>> vreturn;  // 专门用来返回        if(pRoot == NULL)            return vreturn;                vector<int> vline;  // 存储每一行的结点        vector<vector<int>> v;                stack<TreeNode*> level[2];  // 定义两个栈,一个存储奇数行,一个存储偶数行        int current = 0;        int next = 1;                level[current].push(pRoot);          while(!level[0].empty() || !level[1].empty())        {            TreeNode* pNode = level[current].top();            level[current].pop();            vline.push_back(pNode->val);                        if(current == 0)  // 如果当前行是奇数,先把下一行的左结点压入栈,,因为左结点后弹出            {                if(pNode->left != NULL)                   {                    level[next].push(pNode->left);                }                if(pNode->right != NULL)                {                    level[next].push(pNode->right);                }            }            else     // 如果当前行是偶数,先把下一行中的右结点压入栈,,因为右结点后弹出            {                if(pNode->right != NULL)                {                    level[next].push(pNode->right);                }                if(pNode->left != NULL)                {                    level[next].push(pNode->left);                }            }                        if(level[current].empty())            {                v.push_back(vline);                vline.clear();  // 每次都要把 vline 那一行清除,以便存储下一行的数据                current = 1-current;                next = 1-next;            }                    }        return v;    }    };

    分析:按照之字形打印二叉树,奇数行按照从左到右的顺序打印,偶数行按照从右往左的顺序打印。这样,定义两个栈即可,分别存储奇数行和偶数行

阅读全文
1 0
原创粉丝点击