题意:将一个二叉树转换成一个链表,转换的规则:后一个节点是前一个节点在树的先序遍历中的后继节点
方法:把节点分成以下几种类型:
左孩子不为空,无论右孩子是否为空,则该节点的先序遍历的后继节点为该节点的左孩子节点;
左孩子为空,右孩子不为空,则则该节点的先序遍历的后继节点为该节点的右孩子节点;
左孩子、右孩子均为空。首先将进行先序遍历,同时用previous代表前一个被遍历的节点。如果这个节点满足左右孩子均为空,则该节点的右指针指向当前遍历的节点。
然后再从跟开始遍历,若当前节点的左孩子不为空,则让他的右孩子指向左孩子,同时左孩子为空。代码如下:
class Solution {
public:
TreeNode * previous;
void preOrder(TreeNode * temp)
{
if(temp==NULL) return;
if(previous!=NULL && previous->right==NULL&&previous->left==NULL) previous->right=temp;
previous=temp;
preOrder(temp->left);
preOrder(temp->right);
}
void changeLeft(TreeNode * temp)
{
if(temp==NULL) return;
if(temp->left!=NULL)
{
temp->right=temp->left;
temp->left=NULL;
}
changeLeft(temp->right);
}
void flatten(TreeNode* root) {
TreeNode * temp=root;
if(root==NULL) return;
previous=NULL;
preOrder(temp);
temp=root;
changeLeft(temp);
}
};