Populating Next Right Pointers in Each Node II

最新推荐文章于 2021-01-29 16:34:55 发布

a83610312

最新推荐文章于 2021-01-29 16:34:55 发布

阅读量549

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/a83610312/article/details/8922696

本文介绍了一种将任意二叉树转换为链表的层序遍历方法，通过直观处理每层节点的连接，实现树形结构到链表结构的高效转换。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

跟上一题差不多，区别是这次是任意的二叉树。

然后马上反应过来上一题的思路反而有点绕，其实说到底还是很直观的层序遍历，然后遍历的时候处理下一层的连接，连接完成后下一层已经连成一条链表，然后在下沉层序遍历下一层就好了。

代码还是比较容易理解的，就不解释了

/**
 * Definition for binary tree with next pointer.
 * struct TreeLinkNode {
 *  int val;
 *  TreeLinkNode *left, *right, *next;
 *  TreeLinkNode(int x) : val(x), left(NULL), right(NULL), next(NULL) {}
 * };
 */
class Solution {
public:
    void connect(TreeLinkNode* root)
{
    TreeLinkNode* curHead=root;;
    while(curHead)
	{
		TreeLinkNode* cur=curHead;
        TreeLinkNode* nextHead=NULL;
		TreeLinkNode* lastNode=NULL;
		while(cur)
		{
			if (!cur->left&&!cur->right)
            {
                //nothing
            }
			else if (cur->left&&cur->right)
			{
				if (lastNode)
					lastNode->next=cur->left;
				cur->left->next=cur->right;
				lastNode=cur->right;
                if (!nextHead)
                    nextHead=cur->left;
			}
			else
			{
				TreeLinkNode* tmp=cur->left?cur->left:cur->right;
				if (lastNode)
					lastNode->next=tmp;
				lastNode=tmp;
                if (!nextHead)
                    nextHead=tmp;
			}
			cur=cur->next;
		}
        if ( lastNode )
            lastNode->next=NULL;
		curHead=nextHead;
	}
}

};