二叉树分层遍历

首先定义二叉树的存储结构：

　　

struct TreeNode {
    int val;
    TreeNode *left;
    TreeNode *right;

    TreeNode(int v, TreeNode* l = NULL, TreeNode *r = NULL)
        :val(v), left(l), right(r) {}
};

 

1.递归的方法（《编程之美》3.10）

　　二叉树本身就带有递归属性，通常我们可以用递归方法解决。假设要访问第k层节点，那么其实可以转皇城分别访问“以该二叉树根节点的左右子节点为根节点的两棵子树”中层次为k-1的节点。此方法需要求出二叉树的深度，其实可以直接访问到二叉树某一层次失败的时候返回就可以了。

　　这个方法的问题是递归调用，效率较低。而且对每一层的访问都需要从根节点开始，效率极差。

　　最坏的情况下（不平衡树）时间复杂度为O(n^2),空间复杂度O(1)

//输出以root为根节点中的第level层中的所有节点（从左至右），成功返回1
//失败返回0
//root为二叉树根节点
//level为层次数，其中根节点为第0层
int PrintNodeAtLevel(TreeNode *root, int level) {
    if (!root || level < 0) return 0;
    if (level == 0){
        cout<<root->val;
        return 1;
    }

    return PrintNodeAtLevel(root->left, level - 1) + PrintNodeAtLevel(root->right, level - 1);
}

//层次遍历二叉树
//root，二叉树的根节点
void LevelOrder(TreeNode *root) {
    for (int level = 0; ; level++) {
        if (!PrintNodeAtLevel(root, level)) 
            break;
        cout<<endl;
    }
}

 

2. 使用数组和两个游标的方法（《编程之美》 3.10）

　　在访问k层的时候，我们只需要知道k-1层的信息就足够了，所以在访问第k层的时候，要是能够知道k-1层的节点信息，就不再需要从根节点开始遍历了。

　　根据上述分析，可以从根节点出发，依次将每一层的根节点从左往右压入一个数组，并并用一个游标cur记录当前访问的节点，另一个游标last指示当前层次的最后一个节点的下一个位置，以cur===last作为当前层次访问结束的条件，在访问某一层的同时将该层的所有节点的子节点压入数组，在访问完某一层之后，检查是否还有新的层次可以访问，直到检查完所有的层次（不再有新的节点可以访问）

　　这种方法需要一个vector一直存储所有节点，空间效率较差。

　　时间复杂度为O(n),空间复杂度为O(n)

void LevelOrder(TreeNode *root) {
    if (root == NULL) return;
    vector<TreeNode *> vec; //这里使用stl中的vector代替数组，可利用到
                            //其动态扩展的属性
    vec.push_back(root);
    int cur = 0, last = vec.size();
    while (cur < vec.size()) {
        last = vec.size();

        while (cur < last) {
            cout<<vec[cur]->val;
            if (vec[cur]->left) 
                vec.push_back(vec[cur]->left);
            if(vec[cur]->right)
                vec.push_back(vec[cur]->right);
            ++cur;
        }
        cout<<endl;
    }
}

 

3. 两个队列的方法

　　广度优先搜索的思想。使用两个队列，一个记录当前层的节点，另一个记录下一层的节点。输出当前层节点后交换，使下一层的节点称为当前层的节点。

　　时间复杂度O(n),空间复杂度O(1)

void LevelOrder(TreeNode *root) {
    if (root == NULL) return ;

    queue<TreeNode *> current, next;
    
    current.push(root);
    while (!current.empty()) {
        while (!current.empty()) {
            TreeNode * p = current.front();
            cout<<p->val<<" ";
            current.pop();
            if (p->left)
                next.push(p->left);
            if (p->right)
                next.push(p->right);
        }
        cout<<endl;
        swap(next, current);
    }
}

 

4.使用一个队列和两个标记的方法

　　使用current记录当前节点的数量，nextlevel记录下一层节点的数量。当current==0时就将下一层置为当前层。

　　

void LevelOrder(TreeNode *root) {
    if (root == NULL) return;

    queue<TreeNode *> q;
    q.push(root);
    int nextlevel = 0; //记录下一层节点的数量
    int current = 1; //记录当前层节点的数量

    while (!q.empty()) {
        TreeNode *p = q.front();
        q.pop();
        --current;
        cout<<p->val<<" ";

        if (p->left) {
            q.push(p->left);
            ++nextlevel;
        }
        if (p->right) {
            q.push(p->right);
            ++nextlevel;
        }

        if（current == 0) {
            cout<<endl;
            swap(current, nextlevel);
        }
    }
}

 

　　

5. 使用一个队列加一个标记的方法

　　用队列暂存储节点，每当一层节点进入队列，就在最后加入一个空指针表示当前层结束。

void LevelOrder(TreeNode *root) {
    if (root == NULL) return;

    queue<TreeNode *> q;
    q.push(root);
    q.push(0);
    while (!q.empty()) {
        TreeNode *p =q.front();
        q.pop();
        if (p) {
            cout<<p->val<<" ";
            if (p->left)
                q.push(p->left);
            if (p->right)
                q.push(p->right);
            //当发现空指针（结束信号时），要检查队列是够还有节点
            //如果没有的话还插入新的结束信号，则会造成死循环
        } else if (!q.empty()) {
            q.push(0);
            cout<<endl;
        }
    }
}

 

 参考资料：

　　　　　1.《编程之美》

　　　　　2. 《剑指offer》

　　　　　2. http://www.cnblogs.com/miloyip/archive/2010/05/12/binary_tree_traversal.html