题目:
将一棵二叉树按照前序遍历拆解成为一个假链表
。所谓的假链表是说,用二叉树的 right 指针,来表示链表中的 next 指针。
注意事项
不要忘记将左儿子标记为 null,否则你可能会得到空间溢出或是时间溢出。
1
\
1 2
/ \ \
2 5 => 3
/ \ \ \
3 4 6 4
\
5
\
6
思路:
先按照前序遍历将给出的树的每个节点的数值放入队列中,再将队列中的数变成右斜树,即左子树是空,右子树是队列的首个元素,弹出第一个元素依次进行前面的过程,直至队列为空。
代码:
class Solution {
public:
/**
* @param root: a TreeNode, the root of the binary tree
* @return: nothing
*/
std::queue<int> q;
void pre(TreeNode *root)
{
if(root==NULL) return;
q.push(root->val);
pre(root->left);
pre(root->right);
}
void flatten(TreeNode *root) {
// write your code here
pre(root);
if(q.size()==0)
return;
TreeNode *r1=root;
q.pop();
while(!q.empty())
{
r1->left=NULL;
r1->right= new TreeNode(q.front());
r1=r1->right;
q.pop();
}
}
};
感想:
开始第一个自己定义的函数,我返回值是queue<int>,在实现的函数中用q=pre(root),但是在第一个函数中返回不会写。后来就定义了一个全局变量q,这是我最大的收获。