将二叉树拆成链表

来源:互联网 发布:linux打包war文件 编辑:程序博客网 时间:2024/05/29 02:44

问题描述:

将一棵二叉树按照前序遍历拆解成为一个假链表。所谓的假链表是说,用二叉树的 right 指针,来表示链表中的next 指针。

注意事项

不要忘记将左儿子标记为 null,否则你可能会得到空间溢出或是时间溢出。


样例

              1               \     1          2    / \          \   2   5    =>    3  / \   \          \ 3   4   6          4                     \                      5                       \                        6

解题思路:

如果二叉树的左子树不为空时,直接对右子树进行操作。先分别将左子树和右子树变成链表的形式,然后将链表形式的左子树放在根节点的右子树位置,将链表形式的右子树放在新的右子树后面。

代码实现:

/**
 * 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;
        if (root->left==NULL) {
            flatten(root->right);
            return;
        }
       
        TreeNode *tmp_left=root->left;
        TreeNode *tmp_right=root->right;   
        flatten(tmp_left);
        flatten(tmp_right);
        root->left=NULL;
        root->right=tmp_left;
      while(tmp_left->right != NULL)
            tmp_left=tmp_left->right;


        tmp_left->right=tmp_right;
    }
};

解题感悟:

这个题比较麻烦,要用递归,需要比较严谨的思路。

0 0
原创粉丝点击