劍指offer 二叉搜索樹與雙向鏈表

本文轉載自查看原文 2019-08-17 20:00 490 劍指offer

題目描述

輸入一棵二叉搜索樹，將該二叉搜索樹轉換成一個排序的雙向鏈表。要求不能創建任何新的結點，只能調整樹中結點指針的指向。

 1 /*
 2 struct TreeNode {
 3     int val;
 4     struct TreeNode *left;
 5     struct TreeNode *right;
 6     TreeNode(int x) :
 7             val(x), left(NULL), right(NULL) {
 8     }
 9 };*/
10 class Solution {
11 public:
12     TreeNode* Convert(TreeNode* pRootOfTree)
13     {
14         TreeNode *pLastNodeInTree = NULL;
15         TreeConvert(pRootOfTree, pLastNodeInTree);
16         TreeNode *pFirstNodeInTree = pLastNodeInTree;
17         while (pFirstNodeInTree != NULL && pFirstNodeInTree->left != NULL) {
18             pFirstNodeInTree = pFirstNodeInTree->left;
19         }
20         return pFirstNodeInTree;
21     }
22     void TreeConvert(TreeNode* pNode, TreeNode* &pLastNodeInTree) {
23         if (pNode == NULL) {
24             return;
25         }
26         TreeNode* pCurrent = pNode;
27         if (pNode->left != NULL) {
28             TreeConvert(pNode->left, pLastNodeInTree);
29         }
30         pCurrent->left = pLastNodeInTree;
31         if (pLastNodeInTree != NULL)
32             pLastNodeInTree->right = pCurrent;
33         pLastNodeInTree = pCurrent;
34         if (pNode->right != NULL) {
35             TreeConvert(pNode->right, pLastNodeInTree);
36         }
37     }
38 };

免責聲明！

本站轉載的文章為個人學習借鑒使用，本站對版權不負任何法律責任。如果侵犯了您的隱私權益，請聯系本站郵箱yoyou2525@163.com刪除。