问题描述:将一棵二叉树按照前序遍历拆解成为一个假链表
。所谓的假链表是说,用二叉树的 right 指针,来表示链表中的 next 指针。
样例:
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
*/
vector<TreeNode*> temp;
void dfs(TreeNode *root)
{
if(root==NULL)
return;
temp.push_back(root);
dfs(root->left);
dfs(root->right);
}
void flatten(TreeNode *root) {
// write your code here
if(root==NULL)
return;
dfs(root);
int i;
for(i=0;i<temp.size()-1;i++)
{
temp[i]->left=NULL;
temp[i]->right=temp[i+1];
}
temp[i]->left=NULL;
}
};
做题感想:这个题首先想到的是在原树上进行变化,但是自己的想法写的过不去,让同学看了一下,说我逻辑混乱啊。然后看他的借助向量来完成的,我又看了还有用队列来写的,跟向量这个是一个道理了。