特定深度节点链表

给定一棵二叉树，设计一个算法，创建含有某一深度上所有节点的链表（比如，若一棵树的深度为 D，则会创建出 D 个链表）。返回一个包含所有深度的链表的数组。

输入：[1,2,3,4,5,null,7,8]

    1
   /  \ 
  2    3
 / \    \ 
4   5    7

/
8

输出：[[1],[2,3],[4,5,7],[8]]

思路：简单层序遍历

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    vector<ListNode*> listOfDepth(TreeNode* tree) {
		vector<ListNode*>res;
		if (tree==nullptr){
			return res;
		}
		queue<TreeNode*> que;
		que.push(tree);
		while(!que.empty()) {
			int size = que.size();
			ListNode* head = nullptr;
			ListNode* root = nullptr;
			for (int i = 0; i < size; i++)
			{
				TreeNode* temp = que.front();
				que.pop();
				// 从这儿开始
				ListNode* node = new ListNode(temp->val);
				if (head == nullptr) {
					root = node;
				}
				else {
					head->next = node;
				}
				head = node;
				if (temp->left != nullptr) {
					que.push(temp->left);
				}
				if (temp->right != nullptr) {
					que.push(temp->right);
				}
			}
			res.push_back(root);
		}
		return res;
    }
};