链接二叉树中同一层的节点
這個題目之前做過了,但是由于思路不清晰,今天思路又卡了
比如:
? ?? ?? ???0
? ?? ? 1? ?? ? 2
? ?3? ?? ?4? ?? ?? ?5
? ?? ?6? ?? ?? ? 7?
1->2
3->4->5
6->7
參考數(shù)據(jù)結(jié)構(gòu):
struct TREE_NODE
{
? ? ? ? TREE_NODE(int n) : nVal(n), pLft(NULL), pRgt(NULL), pSib(NULL)
? ? ? ? {}
? ? ? ? int nVal;
? ? ? ? TREE_NODE* pLft;
? ? ? ? TREE_NODE* pRgt;
? ? ? ? TREE_NODE* pSib;
};
對于每一個節(jié)點(diǎn)tnode,都要將其左右節(jié)點(diǎn)鏈接。如果沒有右節(jié)點(diǎn),則就要找和tnode同一層的節(jié)點(diǎn),直到找到一個節(jié)點(diǎn)有左孩子或者右孩子,返回這個節(jié)點(diǎn)的孩子。
有一個問題要解決,在找和tnode同層的節(jié)點(diǎn)的孩子節(jié)點(diǎn),必須保證tnode同層的所有節(jié)點(diǎn)已經(jīng)鏈接了,用先左再右的遞歸不行,所以要用從右到左的遞歸。
TREE_NODE *find_next(TREE_NODE *root) {while(root = root->pSib) {if(root->pLft || root->pRgt)return root->pLft?root->pLft:root->pRgt;}return NULL; } void LinkLevelNodes(TREE_NODE *root) {if(!root) return ;if(root->pLft)root->pLft->pSib = root->pRgt?root->pRgt:find_next(root);if(root->pRgt)root->pRgt->pSib = find_next(root);LinkLevelNodes(root->pRgt);LinkLevelNodes(root->pLft); }
總結(jié)
以上是生活随笔為你收集整理的链接二叉树中同一层的节点的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 再谈C++类型转换
- 下一篇: 找出第一个只出现一次的字符