要求: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 }