描述:
将一棵二叉树按照前序遍历拆解成为一个假链表
。所谓的假链表是说,用二叉树的 right 指针,来表示链表中的 next 指针。
样例:
1
\
1 2
/ \ \
2 5 => 3
/ \ \ \
3 4 6 4
\
5
\
6
实现思路:
根据提示,相当于左二子是NULL,我们可以把左子树都存储在右子树里面,通过右指针将所有的节点串联起来,然后遍历整棵树。
实现代码:
/**
* Definition of TreeNode:
* class TreeNode {
* public:
* int val;
* TreeNode *left, *right;
* TreeNode(int val) {
* this->val = val;
* this->left = this->right = NULL;
* }
* }
*/
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;
while (root!=NULL) {
if (root->left!=NULL) {
TreeNode *p = root->left;
while (p->right!=NULL)
p = p->right;
p->right = root->right;
root->right = root->left;
root->left = NULL;
}
root = root->right;
}
}
};
做题感想:通过这个题,进一步体会指针的用法,而且怎样交换两个节点之间的值。