层次遍历二叉树

   

    层次遍历二叉树，编程之美上看过解法，然后在练习了一下。用递归和vector,队列实现了，然后加上了测试函数，测试函数的二叉树创建方法待改进。

//有一棵二叉树，请设计一个算法，按照层次打印这棵二叉树。
//给定二叉树的根结点root，请返回打印结果，结果按照每一层一个数组进行储存，所有数组的顺序按照层数从上往下，且每一层的数组内元素按照从左往右排列。保证结点数小于等于500。

#include <iostream>
using namespace std;
#include <vector>
#include<deque>
#define N 10

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

class TreePrinter {
public:
    vector<vector<int> > printTree(TreeNode* root) {
        // write code here
        vector<vector<int> > vc;
        for (int level = 0;; level++)
        {
            if (!printTreeLevel(root, level))  //巧妙地避开了访问二叉树的深度
            {
                break;
            }
            vc.push_back(res);
            res.clear();
        }
        return vc;
    }

    int  printTreeLevel(TreeNode* root, int level)//访问以root为根节点的第level层结点。失败返回0，成功返回1；
    {
        if (!root)  //只有5这个结点有孩子没有置空
        {
            return 0;
        }
        if (level == 0)
        {
            res.push_back(root->val);
            return 1;
        }
        return printTreeLevel(root->left, level - 1) + printTreeLevel(root->right, level - 1);
    }

    //二叉树的深度
    int deepth(TreeNode* root)
    {
        if (!root)
        {
            return 0;
        }
        return deepth(root->left) > deepth(root->right) ? (deepth(root->left) + 1) : (deepth(root->right) + 1);
    }
private:
    vector<int> res;

    //方法二:从根节点出发，依次将每层结点从左到右压入一个数组，并用cur记录当前访问的结点，last标记当前层的最后一个结点的下一个位置（下标从0开始），当cur==last表示当前层访问结束
public:
    //vector<vector<int> > printTree2(TreeNode* root)
    //{
    //    vector<vector<int> > res;
    //    if (!root)
    //    {
    //        return res;
    //    }
    //    vector<TreeNode*> vc;
    //    vc.push_back(root);
    //    vector<int> level;
    //    int cur = 0;
    //    int last = 1;
    //    while (cur < vc.size())
    //    {
    //        last = vc.size();
    //        while (cur < last)
    //        {
    //            level.push_back(vc[cur]->val);
    //            if (vc[cur]->left)
    //            {
    //                vc.push_back(vc[cur]->left);
    //            }
    //            if (vc[cur]->right)
    //            {
    //                vc.push_back(vc[cur]->right);
    //            }
    //            cur++;
    //        }
    //        res.push_back(level);
    //        level.clear();
    //    }
    //    return res;
    //}

    ////运用队列
    //vector<vector<int> > printTree1(TreeNode* root) {
    //    vector<vector<int> > result;
    //    if (root == NULL) 
    //        return result;
    //    deque<TreeNode*> q;
    //    TreeNode* last = root;
    //    TreeNode* nlast = root;  //指向下一层的最后一个结点
    //    vector<int> tmp;
    //    q.push_back(root);
    //    while (!q.empty()){
    //        TreeNode* cur = q.front();
    //        q.pop_front();
    //        tmp.push_back(cur->val);
    //        if (cur->left) {
    //            q.push_back(cur->left);
    //            nlast = cur->left;
    //        }
    //        if (cur->right) {
    //            q.push_back(cur->right);
    //            nlast = cur->right;
    //        }
    //        if (last == cur) {
    //            result.push_back(tmp);
    //            tmp.clear();
    //            last = nlast;
    //        }
    //    }
    //    return result;
    //}
    //private:
    //    char test[] ;
    //void createBiTree(TreeNode* root)  //字符串类型的树创建方法
    //{
    //    int i = 0;
    //    char c;
    //    if (test[i]&&'#'==(c==test[i]))
    //    {
    //        root = nullptr;
    //    }
    //    else
    //    {
    //        root = new TreeNode(1);
    //        root->val = c;
    //        createBiTree(root->left);
    //        createBiTree(root->right);
    //    }
    //}

    //数字类型的树创建的方法   //所以这样方式创建还需要改进
    //生成一个结点
    TreeNode* createNode(int i)
    {
        TreeNode* q = new TreeNode(i);
        q->left = nullptr;
        q->right = nullptr;
        return q;
    }
    TreeNode* creteTree()
    {
        TreeNode* p[N]={nullptr };
        for (int i = 0; i < N;i++)
        {
            p[i] = createNode(i+1);
        }
        for (int i = 0; i < N / 2;i++)
        {
            p[i]->left = p[i * 2 + 1];
            p[i]->right = p[i * 2 + 2];
        }
        p[4]->right = nullptr;
        return p[0];
    }

public:
    int getLeafNode(TreeNode* T)
    {
        if (NULL == T)
            return 0;

        if (NULL == T->left && NULL == T->right)
            return 1;

        return getLeafNode(T->left) + getLeafNode(T->right);
    }
};

int main()
{
    TreePrinter t;
    TreeNode* T;
    T=t.creteTree();
    vector<vector<int> > vc = t.printTree(T);
    for (int i = 0; i < vc.size();i++)
    {
        for (int j = 0; j < vc[i].size();j++)
        {
            cout << vc[i][j]<<" ";
        }
        cout << endl;
    }
    return 0;
}