Given a binary tree, flatten it to a linked list in-place.
For example,
Given
1 / 2 5 / \ 3 4 6
The flattened tree should look like:
1 2 3 4 5 6
思路:
用先序遍历,得到的是从小到大的顺序。把先序遍历变形一下:
void flatten(TreeNode* root) { if(NULL == root) return; vector<TreeNode *> v; v.push_back(root); TreeNode * pCur = NULL; TreeNode * p = NULL; while(!v.empty()) { pCur = p = v.back(); while(NULL != p->left) //向左移动时只要把当前指针移动即可,因为是按照从小大的顺序的 { v.push_back(p->left); pCur = p = p->left; } while(!v.empty() && (p = v.back()->right) == NULL) //找到下一个要处理的元素,一定是一个右子树 v.pop_back(); if(!v.empty()) v.pop_back(); if(p != NULL) //把待处理的右子树连在当前已经铺平的数的左子树上 { v.push_back(p); pCur->left = p; } } p = root; while(p != NULL) //镜像,全部转移到右子树上 { p->right = p->left; p->left = NULL; p = p->right; } }
我发现,我太依赖非递归了,其实有的时候用递归更好。比如大神的版本:
private TreeNode prev = null; public void flatten(TreeNode root) { if (root == null) return; flatten(root.right); flatten(root.left); root.right = prev; root.left = null; prev = root; }
先把右子树放平,再把左子树放平。因为每次都先处理右子树,所以prev是数值从大到小的一个记录,每次prev都是恰比当前根节点大一点的那个指针的位置。(不大好理解)
下面这个,跟上面的思路很像,但是会容易理解很多:
public void flatten(TreeNode root) { if (root == null) return; TreeNode left = root.left; TreeNode right = root.right; root.left = null; flatten(left); flatten(right); root.right = left; TreeNode cur = root; while (cur.right != null) cur = cur.right; cur.right = right; }
也是先铺平右子树,再铺平左子树。一定是右子树的值大于左子树,所以先把左子树连接到根节点右侧,再把右子树连在后面。 这个代码平铺左右子树的顺序没限制。反过来也行。
上面思路的非递归版本: 也是需要想半天才能看懂的。每次先把紧接着子树连好,其他代码都是从叶节点开始连接,这个是从根节点开始的。
public void flatten(TreeNode root) { if (root == null) return; Stack<TreeNode> stk = new Stack<TreeNode>(); stk.push(root); while (!stk.isEmpty()){ TreeNode curr = stk.pop(); if (curr.right!=null) stk.push(curr.right); if (curr.left!=null) stk.push(curr.left); if (!stk.isEmpty()) curr.right = stk.peek(); curr.left = null; // dont forget this!! } }
时间: 2024-10-16 04:16:58