【二叉树】创建含有某一深度上所有结点的链表

实现功能

给定一棵二叉树，设计一个算法，创建含有某一深度上所有结点的链表。比如，若一棵树的深度为D，则会创建出D个链表。

功能代码

/********************************************
 *  Author        : Dong Huaan
 *  Email         : donghuaan@163.com
 *  Filename      : createLinkedListForBinaryTree.cpp
 *  Creation time : 2018-01-03 14:13:44
 *  Last modified : 2018-01-03 15:03:44
 *  Description   : ---
 *******************************************/

#include<iostream>
#include<vector>
#include<list>
#include"BinaryTree.h"

// 利用已经构建出来的链表进行广度优先遍历。
std::vector<std::list<BTNode*>*> * createLinkedListForBinaryTree(BTNode *root)
{
    if(!root) return nullptr;

    std::vector<std::list<BTNode*>*> * vector = new std::vector<std::list<BTNode*>*>;
    BTNode *temp = root;
    int index = 0;
    std::list<BTNode*> *list = new std::list<BTNode*>;
    vector->push_back(list);
    (*vector)[index]->push_back(temp);

    for(; !(*vector)[index]->empty(); ++index)
    {
        list = new std::list<BTNode*>;
        vector->push_back(list);
        for(std::list<BTNode*>::iterator iter = (*vector)[index]->begin(); iter != (*vector)[index]->end(); ++iter)
        {
            temp = *iter;
            if(temp->left) (*vector)[index + 1]->push_back(temp->left);
            if(temp->right) (*vector)[index + 1]->push_back(temp->right);
        }
    }

    vector->pop_back();

    delete list;

    return vector;
}

测试

功能测试：进行了少量用例的功能测试，未进行大量用例测试。
边界值测试：未进行边界值测试。
特殊输入测试：未进行特殊输入测试。

测试代码

void traverseLinkedLists(std::vector<std::list<BTNode*>*> * vector)
{

    for(std::vector<std::list<BTNode*>*>::iterator iter = vector->begin(); iter != vector->end(); ++iter)
    {
        std::list<BTNode*> *list = *iter;
        for(std::list<BTNode*>::iterator it = list->begin(); it != list->end(); ++it)
            std::cout << (*it)->data << "  ";
        std::cout << "\n";
    }
}

void test()
{
        BTNode *root = nullptr;

    std::cout << "create binary tree:\n";
    createBinaryTree(root);

    std::cout << "\npre order traverse binary tree:\n";
    preOrdertraverseBinaryTree_Recur(root);

    std::vector<std::list<BTNode*>*> *result = createLinkedListForBinaryTree(root);

    std::cout << "\n\ntraverse linked lists:\n";

    traverseLinkedLists(result);
}

int main()
{
    test();
}

测试结果