要求:Given a binary tree, flatten it to a linked list in-place.将二叉树转化为平坦序列的树。比如:
结题思路:
该题有个提示,转化后的树的序列正好是二叉树前序遍历所得到的序列,所以,该题第一个思路就是利用前序遍历的方式来做。
第二个思路:我们可以利用递归的思路,先对根节点进行处理,将root的左子树放到右子树,在将左子树中的最右端节点“嫁接”到右子树,接着上面得到的子树。接下来在用同样的方法处理当前二叉树的下一个节点。看下面一个图,即可明了。
我实现的是第二种思路,代码如下:
1 struct TreeNode { 2 int val; 3 TreeNode* left; 4 TreeNode* right; 5 TreeNode(int x): val(x), left(NULL),right(NULL) {} 6 }; 7 8 void flatten(TreeNode *root) 9 { 10 if (NULL == root) 11 return; 12 //使用stack的非递归方法 13 TreeNode *left = root->left; 14 TreeNode *right = root->right; 15 16 if (left) { 17 root->right = left; 18 root->left = NULL; 19 20 TreeNode *pTemp = left; 21 while (pTemp->right) 22 pTemp = pTemp->right; 23 pTemp->right = right; 24 25 flatten(root->right); 26 } 27 }