leetCode_Flatten Binary Tree to Linked List

来源:互联网 发布:手机上传淘宝主图视频 编辑:程序博客网 时间:2024/06/06 05:40

题意:将一个二叉树转换成一个链表,转换的规则:后一个节点是前一个节点在树的先序遍历中的后继节点

方法:把节点分成以下几种类型:

左孩子不为空,无论右孩子是否为空,则该节点的先序遍历的后继节点为该节点的左孩子节点;

左孩子为空,右孩子不为空,则则该节点的先序遍历的后继节点为该节点的右孩子节点;

左孩子、右孩子均为空。首先将进行先序遍历,同时用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);    }};


0 0
原创粉丝点击