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