Leetcode#117 Populating Next Right Pointers in Each Node II

原题地址

 

二叉树的层次遍历。

对于每一层，依次把各节点连起来即可。

 

代码：

 1 void connect(TreeLinkNode *root) {
 2   if (!root) return;
 3 
 4   queue<TreeLinkNode *> parents;
 5 
 6   parents.push(root);
 7   while (!parents.empty()) {
 8     queue<TreeLinkNode *> children;
 9 
10     while (!parents.empty()) {
11       TreeLinkNode *p = parents.front();
12       parents.pop();
13       p->next = parents.empty() ? NULL : parents.front();
14       if (p->left)
15         children.push(p->left);
16       if (p->right)
17         children.push(p->right);
18     }
19     parents = children;
20   }
21 }