主要进行先序遍历,把指针都存到数组里(这样不会改变指针的指向结构)
然后遍历数组,按顺序更新关系
/*
114. 二叉树展开为链表
给你二叉树的根结点 root ,请你将它展开为一个单链表:
展开后的单链表应该同样使用 TreeNode ,其中 right 子指针指向链表中下一个结点,而左子指针始终为 null 。
展开后的单链表应该与二叉树 先序遍历 顺序相同。
示例 1:
输入:root = [1,2,5,3,4,null,6]
输出:[1,null,2,null,3,null,4,null,5,null,6]
示例 2:
输入:root = []
输出:[]
示例 3:
输入:root = [0]
输出:[0]
*/
class Solution {
public:
vector<TreeNode*> temp;
void dfs(TreeNode*root,vector<TreeNode*> &temp){
if(root){
temp.push_back(root);
dfs(root->left,temp);
dfs(root->right,temp);
}
}
void flatten(TreeNode* root) {
dfs(root,temp);
for(int i=1;i<temp.size();i++){
//实例化两个指针
TreeNode* pre;
TreeNode* curr;
pre=temp[i-1];
curr=temp[i];
//构建关系
pre->left=nullptr;
pre->right=curr;
}
}
};