Lintcode---将二叉树拆成链表

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

 注意事项

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

样例
              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
     */
    /*
    思路:采用递归的方法,按照先序遍历,先存放起来;
          但是,要注意将左儿子标记为NULL;
    */
    
    vector<TreeNode*> vec;
    vector<TreeNode*> Node(TreeNode* root,vector<TreeNode*>& vec){
        if(root==NULL){
            return vec;
        }
        vec.push_back(root);
        if(root->left!=NULL){
            Node(root->left,vec);
        }
        if(root->right!=NULL){
            Node(root->right,vec);
        }
        return vec;
    }
    
    void flatten(TreeNode *root) {
        // write your code here
        if(root==NULL){
            return;
        }
        
        Node(root,vec);
        root=new TreeNode();//创建新的头结点指针;
        for(int i=0;i<vec.size();i++){
            //多想想,第一次的写法为啥没有通过
            vec[i]->left=NULL;
            root->right=vec[i];
            root=root->right;
        }
    }
};

 



posted @ 2017-06-22 19:05  静悟生慧  阅读(426)  评论(0编辑  收藏  举报