二叉樹層序遍歷應用:要求將所有結點加上一個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萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章