第一次做LeetCode,选了一道简单一点的题目,先熟悉一下环境。
题目如下:
Given a binary tree
struct TreeLinkNode { TreeLinkNode *left; TreeLinkNode *right; TreeLinkNode *next; }
Populate each next pointer to point to its next right node. If there is no next right node, the next pointer should be set to NULL
.
Initially, all next pointers are set to NULL
.
Note:
- You may only use constant extra space.
- You may assume that it is a perfect binary tree (ie, all leaves are at the same level, and every parent has two children).
For example,
Given the following perfect binary tree,
1 / \ 2 3 / \ / \ 4 5 6 7
After calling your function, the tree should look like:
1 -> NULL / \ 2 -> 3 -> NULL / \ / \ 4->5->6->7 -> NULL
class Solution {
public:
void connect(TreeLinkNode *root) {
for(;root!=NULL;root=root->left)
{
createConnect(root->left,root);
}
}
void createConnect(TreeLinkNode *dest,TreeLinkNode *parent)
{
if(dest==NULL) return;
TreeLinkNode *tmpP,*tmpD;
dest->next=parent->right;
for(tmpP=parent->next,tmpD=dest->next;tmpP!=NULL;tmpP=tmpP->next,tmpD=tmpD->next->next)
{
tmpD->next=tmpP->left;
tmpD->next->next=tmpP->right;
}
}
};
程序很简单,没啥算法可言。由上到下,循环遍历每个层,遍历的时候带着父辈的节点,因为父辈节点已经形成了链式结构,所以挨个给该层节点的next赋值就行了。
说说碰到的问题:
1、当然是输入NULL节点,不考虑的话,会报Runtime Error。
2、只有一个节点,一样,考虑好节点各个指针的NULL情况。
3、在连接完第一个父节点的两个孩子后,没有把当前处理的节点后移,导致在层数 >=4的情况下,中间的节点只会内部连接起来,而不会连接到整个链上(因为被跳过了)。最后只有父节点为上一层第一个和最后一个的孩子节点被连接起来。
PS. csdn的编辑器,也太难用了吧。。