Q4.4 每一层的所有结点构建为一个链表

最新推荐文章于 2024-09-14 18:52:56 发布

weixin_34186128

最新推荐文章于 2024-09-14 18:52:56 发布

阅读量81

点赞数

文章标签：数据结构与算法

原文链接：http://www.cnblogs.com/xuanxu/p/3243221.html

版权

Given a binary search tree, design an algorithm which creates a linked list of all the nodes at each depth (i.e., if you have a tree with depth D, you’ll have D linked lists).

#include <list>

typedef struct Node
{
    int val;
    Node *left,*right;
}

vector<list<Node*>> findLevelList(Node *root)
{    int level = 0;
    vector<list<Node*>>     res;
    list<Node*> aList;
    aList.push_back(root);
    res.push_back();
    
    while(!res[level].empty())
    {    list<Node*> li;
        list<Node*>::iterator itr;
        for(itr = res[level].begin;
            itr!=res[level].end();itr++)
            {    Node *n = *itr;
                if(n->left) li.push_back(n->left);
                if(n->right)li.push_back(n->right);
            }
        ++level;
        res.push_back(li);
    }
    return res;
}