题目描述:
将一棵二叉树按照前序遍历拆解成为一个假链表
。所谓的假链表是说,用二叉树的right 指针,来表示链表中的 next 指针。
注意事项
不要忘记将左儿子标记为 null,否则你可能会得到空间溢出或是时间溢出。
样例:
1
\
1 2
/ \ \
2 5 => 3
/ \ \ \
3 4 6 4
\
5
\
6
做题思路:运用递归遍历二叉树。如果根结点为空,跳出递归,如果左孩子为空,则遍历右子树。建立两个新的结点,把根结点的左右孩子的值赋给它们,用递归函数遍历,让左孩子为空,把原来左孩子的值赋给右孩子,当该结点的右孩子不为空时,再把左子树的结点放进右子树中,实现链表。
关键代码:
class Solution {
public:
/**
* @param root: a TreeNode, the root of the binary tree
* @return: nothing
*/
void flatten(TreeNode *root) {
// write your code here
if(root==NULL) return;
if(root->left==NULL)
{ flatten(root->right);
return;
}
TreeNode *left=root->left;
TreeNode *right=root->right;
flatten(left);
flatten(right);
root->left=NULL;
root->right=left;
while(left->right!=NULL)
left=left->right;
left->right=right;
}
};
做题感想:这题自己做的时候总是wa和超时,上网搜答案一直没看懂,后来在纸上画了好几遍,明白了这题的思路。