问题描述:
将一棵二叉树按照前序遍历拆解成为一个假链表
。所谓的假链表是说,用二叉树的 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*> tree;
void PreorderTraversal(TreeNode *root) {
if (root== NULL)return;
tree.push_back(root);
PreorderTraversal(root->left);
PreorderTraversal(root->right);
}
void Flatten(TreeNode *Root, int i) {
Root->left= NULL;
Root->right= new TreeNode(0);
Root->right= tree[i];
i++;
if (i== tree.size()+1) {
Root->right= NULL;
Root->left= NULL;
return;
}
Flatten(Root->right, i);
}
void flatten(TreeNode *root) {
// write your code here
if (root== NULL)return;
PreorderTraversal(root);
int j= 0;
Flatten(root, j);
}
};
感想:
还是考察的遍历 ,外加树的存储。