二叉树层序遍历应用:要求将所有结点加上一个m_pNext指针,指向同层右侧的结点,右侧若无结点则指向NULL,写出其中的Connect函数。

该二叉树结点结构:

struct TreeNode{
  TreeNode* m_pLeft;
  TreeNode* m_pRight;
  TreeNode* m_pSibling;
  int m_nValue;
}
层序遍历主要是利用队列。

Connect函数的代码如下:

TreeNode* Connect(TreeNode* root)
{
   	if(root==NULL)
		return	NULL; 
	std::deque< TreeNode*>queueNode;
	std::deque< TreeNode*>iterator iter;

	queueNode.push_back(root);
	while(!queueNode.empty())
	{
		for(iter=queueNode.begin;iter!=queueNode.end-1;iter++)
		{
			(*iter)->m_pSibling=*(iter+1);
		}
		(*iter)->m_pSibling=NULL;
		for(iter=queueNode.begin;iter!=queueNode.end;iter++)
		{
			queueNode.push_back((*iter)->m_pLeft);
			queueNode.push_back((*iter)->m_pRight);
			queueNode.pop_front();
		}
	}
	return root;
}
发布了22 篇原创文章 · 获赞 13 · 访问量 12万+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章