给定一个二叉树,原地将它展开为链表。
例如,给定二叉树
1 / \ 2 5 / \ \ 3 4 6
将其展开为:
1 \ 2 \ 3 \ 4 \ 5 \ 6
解题思路:
最近做了较多的二叉树题目,感觉使用递归的话会变得贼简单,其实细想一下二叉树的结构,有左子树,右子树,在结构上本质和根节点包含的树并没有什么区别,处理成递归是结构所导致。对于简单的递归,我们可以考虑递归消除,这样既能保证计算速度,又能不影响代码的阅读效率;然而对于比较浮渣的递归而言,强行递归消除可能得到了一时的速度优势,但是代码不方便阅读极可能影响后续的开发。当然以上只是个人看法。
回归到本题的解题思路:
1. 本题的问题是将一颗树转换成链表,其实这个链表并非真正的链表,本质还是树,我在这称作链表树!只不过对于任意的节点只有右孩子,正如例子给出的效果那样。
2. 我的方法是,递归,将左子树变成链表树,根为root->left,返回叶子结点LEFT_end。将右子树变成链表树root->right,返回叶子结点RIGHT_end。
3.连接,根结点root,以及这两个树。
注意,这两个树都可能不存在!
#define hasLChild(x) (!(x->left==NULL)) #define hasRChild(x) (!(x->right==NULL)) #define hasNoChild(x) (!hasLChild(x)&&!hasRChild(x)) class Solution { public: void flatten(TreeNode* root) { if (root == NULL) return ; DFS(root); } TreeNode* DFS(TreeNode* root) {//返回链表表尾 if (hasNoChild(root)) return root; TreeNode* LTREE_begin = root->left, *LTREE_end = NULL, *RTREE_begin = root->right, *RTREE_end = NULL; //获取两个链表树,如果有的话 if (hasLChild(root)) LTREE_end = DFS(root->left); if (hasRChild(root)) RTREE_end = DFS(root->right); //链接它们 if (hasLChild(root)) { if (hasRChild(root)) { root->right = LTREE_begin; LTREE_end->right = RTREE_begin; } else { root->right = new TreeNode(0); root->right = root->left; RTREE_end = LTREE_end; } root->left = NULL; } return RTREE_end; } }; |