Flatten Binary Tree to Linked List

原題鏈接Flatten Binary Tree to Linked List

給定一個二叉樹，將其原地平鋪成鏈式結構(按先序遍歷的方式平鋪)

雖然結果是先序遍歷，但是本題不能直接在二叉樹上進行先序遍歷操作，原因是題目要求在原內存上做修改，也就是說只能改變節點的left和right指針

遇到二叉樹的問題，首先想到的就是遞歸，所以本題仍然需要利用遞歸求解

對于某個節點，假設它的左右子樹都已經按要求轉換完畢，那么只需要將這兩個子樹拼接在一起即可

代碼如下

/*** Definition for a binary tree node.* struct TreeNode {* int val;* TreeNode *left;* TreeNode *right;* TreeNode(int x) : val(x), left(NULL), right(NULL) {}* };*/ class Solution { public:void flatten(TreeNode* root) {flattenTree(root); } private:/* 返回以root為根節點的子樹平鋪后的樹根節點 */TreeNode* flattenTree(TreeNode* root){if(!root) return nullptr;/* 轉換左子樹 */TreeNode* leftTree = flattenTree(root->left);/* 轉換右子樹 */TreeNode* rightTree = flattenTree(root->right);/* 找到左子樹的最后一個節點 */TreeNode* leftLastNode = leftTree;while(leftLastNode && leftLastNode->right)leftLastNode = leftLastNode->right;/* 將左子樹從根節點中移除(因為要移動到右邊) */root->left = nullptr;/* 只有當左子樹不為空時才需要轉換，當為空時，右子樹位置不需要變化 */if(leftTree){/* 根節點的左子樹作為新的右子樹 */root->right = leftTree;/* 左子樹連接右子樹 */leftLastNode->right = rightTree;}/* 返回當前根節點 */return root;} };

---

本題還是利用遞歸，以后遇到二叉樹的問題，最好首先考慮遞歸求解

總結

以上是生活随笔為你收集整理的每天一道LeetCode-----将二叉树原地平铺成链式结构的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。