
解题思路:
\qquad题目中已经明确,要用先序遍历的顺序展开,那么就需要按照“中-左-右”的顺序遍历二叉树,剩下的问题在于如何在遍历过程中完成链表指针的更新。
\qquad由于“右子节点”需要链接到左子树最后一个节点之后,所以需要额外的指针last记录已展开链表的末尾。由于链表按照前序遍历的方式展开的,所以last即为每次遍历的root。
\qquad每一次遍历的节点root,都需要对链表进行展开,具体将左子树移动到root->right,并对左子树展开,展开完成后,将原右半部分添加到当前遍历的末尾last继续展开。
TreeNode* last = nullptr;
void flatten(TreeNode* root) {
if(root == nullptr) return;
last = root;
TreeNode* n_right = root->right;
if(root->left == nullptr && root->right == nullptr) return;
root->right = root->left;
root->left = nullptr;
flatten(root->right);
last->right = n_right;
flatten(n_right);
}


731

被折叠的 条评论
为什么被折叠?



