Convert Sorted List to Binary Search Tree
阿新 • • 發佈:2019-01-30
Given a singly linked list where elements are sorted in ascending order, convert it to a height balanced BST.
想了好久想不出來,後來看了題目分類裡面說是DFS,但是沒有想出DFS的演算法來。後來想到了一個遞迴的方法,但是空間和時間都是O(n)。
後來網上找了一個空間是O(1)的時間是O(n)的演算法,是一種新的解題思路,用的是中遞迴。一般我解題都是用的頭遞迴或者尾遞迴,第一次見識到了中遞迴,相當於把遞迴當成了一個迴圈體,用引用來作為變數,每個遞迴中修改,需要很強大的想象力,把整個遞迴樹在腦子裡想清楚。
空間為O(1)時間為O(n):
想了好久想不出來,後來看了題目分類裡面說是DFS,但是沒有想出DFS的演算法來。後來想到了一個遞迴的方法,但是空間和時間都是O(n)。
後來網上找了一個空間是O(1)的時間是O(n)的演算法,是一種新的解題思路,用的是中遞迴。一般我解題都是用的頭遞迴或者尾遞迴,第一次見識到了中遞迴,相當於把遞迴當成了一個迴圈體,用引用來作為變數,每個遞迴中修改,需要很強大的想象力,把整個遞迴樹在腦子裡想清楚。
空間和時間都為O(n):
TreeNode *sortedListToBST(ListNode *head) { vector<TreeNode*> treeNodes; while (head != NULL) { TreeNode *node = new TreeNode(head->val); treeNodes.push_back(node); head = head->next; } return genBST(0, treeNodes.size()-1, treeNodes); } TreeNode* genBST(int start, int end, vector<TreeNode*> &treeNodes) { if (start == end) return treeNodes[start]; else if (start+1 == end) { treeNodes[start]->right = treeNodes[end]; return treeNodes[start]; } int mid = (start+end)/2; TreeNode* root = treeNodes[mid]; root->left = genBST(start, mid-1, treeNodes); root->right = genBST(mid+1, end, treeNodes); return root; }
空間為O(1)時間為O(n):
解法引用:http://www.bwscitech.com/a/jishuzixun/javayuyan/2013/0930/15822.htmlTreeNode *sortedListToBST(ListNode *head) { int len = 0; ListNode * node = head; while (node != NULL) { node = node->next; len++; } return buildTree(head, 0, len-1); } TreeNode *buildTree(ListNode *&node, int start, int end) { if (start > end) return NULL; int mid = start + (end - start)/2; TreeNode *left = buildTree(node, start, mid-1); TreeNode *root = new TreeNode(node->val); root->left = left; node = node->next; root->right = buildTree(node, mid+1, end); return root; }