将二叉树拆成链表

来源:互联网 发布:达内软件科技 编辑:程序博客网 时间:2024/06/07 05:20

描述:

     将一棵二叉树按照前序遍历拆解成为一个假链表。所谓的假链表是说,用二叉树的 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;
        }  
        }
};

做题感想:通过这个题,进一步体会指针的用法,而且怎样交换两个节点之间的值。

0 0
原创粉丝点击