剑指刷题 把二叉树按照行打印

题目描述

从上到下按层打印二叉树，同一层结点从左至右输出。每一层输出一行。

解题思路：

使用 层次遍历的方法，将 树的节点类型进行再封装， 成员变量包含树的节点，和该结点处于 树的第几层，  根节点算第1层

然后层次遍历时 根据 该结点 的层 来决定 压入二维数组的哪一维。

 

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
*/

typedef struct ceng_tree{
    TreeNode * tree;
    int ceng; //存放 该结点 属于树的第几 层数  树从 1层开始算起
    ceng_tree( TreeNode * root,int k) 
    {
        tree=root;
        ceng=k;
    }
}ceng_tree;
#include <queue>
class Solution {
public:
        vector<vector<int> > Print(TreeNode* pRoot) {
            vector<vector<int> > res;
            if(pRoot==NULL)
                return res;
            queue<ceng_tree*> q;
            ceng_tree *root =new ceng_tree(pRoot,1);  //使用层次遍历的方法
            q.push(root);
            while(!q.empty())
            {
                root=q.front();
                q.pop();
                if(root->ceng > res .size())  //如果该层数比 结果的大小大，说明这一层还没有被压入结果二维数组中
                {
                    vector<int> temp;
                    temp.push_back(root->tree->val);
                    res.push_back(temp);
                }
                else{ //否则   二维数组的 第层数-1 维  压入 这个结点的val
                    res[root->ceng-1].push_back(root->tree->val);
                }
                if(root->tree->left!=NULL)
                { //构造新的 左子树的 结点 压入 队列中
                    ceng_tree *temp=new ceng_tree(root->tree->left,root->ceng+1); 
                    q.push(temp);
                }
                if(root->tree->right!=NULL)
                {
                    ceng_tree *temp=new ceng_tree(root->tree->right,root->ceng+1);
                    q.push(temp);
                }
            }
            return res;
        }
    
};

题目描述

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

 

解题思路：  首先按照层数 放到二维数组中，然后 将奇数的层进行反转。 

 

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
*/


typedef struct ceng_tree{
    TreeNode * tree;
    int ceng;
    ceng_tree( TreeNode * root,int k)
    {
        tree=root;
        ceng=k;
    }
}ceng_tree;
#include <queue>

class Solution {
public:
    vector<vector<int> > Print(TreeNode* pRoot) {
        vector<vector<int> >  res=Print_ceng(pRoot);
        int depth=res.size();
        for(int i=1;i<depth;i+=2)
        {
            int len=res[i].size();
            int j,k;
            for(j=0,k=len-1;j<k;++j,--k)
            {
                swap(res[i][j],res[i][k]);
            }
        }
        return res;
    }
    vector<vector<int> > Print_ceng(TreeNode* pRoot) {//这是之前按层打印二叉树的题
            vector<vector<int> > res;
            if(pRoot==NULL)
                return res;
            queue<ceng_tree*> q;
            ceng_tree *root =new ceng_tree(pRoot,1);
            q.push(root);
            while(!q.empty())
            {
                root=q.front();
                q.pop();
                if(root->ceng > res .size())
                {
                    vector<int> temp;
                    temp.push_back(root->tree->val);
                    res.push_back(temp);
                }
                else{
                    res[root->ceng-1].push_back(root->tree->val);
                }
                if(root->tree->left!=NULL)
                {
                    ceng_tree *temp=new ceng_tree(root->tree->left,root->ceng+1);
                    q.push(temp);
                }
                if(root->tree->right!=NULL)
                {
                    ceng_tree *temp=new ceng_tree(root->tree->right,root->ceng+1);
                    q.push(temp);
                }
            }
            return res;
        }
};