解题思路:
\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);}