後序遍歷的最後一個節點是根節點,在中序遍歷中找到根節點所在的位置,將該vector分成左右兩部分,然後在後序中找到和它們對應的部分,遞迴求出左右子樹。
1 TreeNode *buildTree(vector<int> &inorder, vector<int> &postorder) { 2 // Start typing your C/C++ solution below 3 // DO NOT write int main() function 4 TreeNode *root = new TreeNode(0); 5 if(inorder.size() == 0){ 6 return NULL; 7 } 8 vector<int> leftInorder, leftPostorder, rightInorder, rightPostorder; 9 int len = inorder.size(); 10 root->val = postorder[len-1]; 11 vector<int>::iterator i; 12 for(i = inorder.begin(); i < inorder.end(); i++){ 13 if(*i == root->val) 14 break; 15 } 16 leftInorder.assign(inorder.begin(), i); 17 rightInorder.assign(i+1, inorder.end()); 18 leftPostorder.assign(postorder.begin(), postorder.begin()+(i-inorder.begin())); 19 rightPostorder.assign(postorder.begin()+(i-inorder.begin()), postorder.end()-1); 20 vector<int>().swap(inorder); 21 vector<int>().swap(postorder); 22 root->left = buildTree(leftInorder, leftPostorder); 23 root->right = buildTree(rightInorder, rightPostorder); 24 return root; 25 }
因為新建了許多vector,所以出現了Memory Limit Exceeded的問題。查了一些資料後,增加了20、21行,解決。關於vector如何釋放記憶體的方法援引如下:
1. vector容器的記憶體自增長
與其他容器不同,其記憶體空間只會增長,不會減小。先來看看"C++ Primer"中怎麼說:為了支援快速的隨機訪問,vector容器的元素以連續方式存放,每一個元素都緊挨著前一個元素儲存。設想一下,當vector新增一個元素時,為了滿足連續存放這個特性,都需要重新分配空間、拷貝元素、撤銷舊空間,這樣效能難以接受。因此STL實現者在對vector進行記憶體分配時,其實際分配的容量要比當前所需的空間多一些。就是說,vector容器預留了一些額外的儲存區,用於存放新新增的元素,這樣就不必為每個新元素重新分配整個容器的記憶體空間。
關於vector的記憶體空間,有兩個函式需要注意:size()成員指當前擁有的元素個數;capacity()成員指當前(容器必須分配新儲存空間之前)可以儲存的元素個數。reserve()成員可以用來控制容器的預留空間。vector另外一個特性在於它的記憶體空間會自增長,每當vector容器不得不分配新的儲存空間時,會以加倍當前容量的分配策略實現重新分配。例如,當前capacity為50,當新增第51個元素時,預留空間不夠用了,vector容器會重新分配大小為100的記憶體空間,作為新連續儲存的位置。
2. vector記憶體釋放
由於vector的記憶體佔用空間只增不減,比如你首先分配了10,000個位元組,然後erase掉後面9,999個,留下一個有效元素,但是記憶體佔用仍為10,000個。所有記憶體空間是在vector析構時候才能被系統回收。empty()用來檢測容器是否為空的,clear()可以清空所有元素。但是即使clear(),vector所佔用的記憶體空間依然如故,無法保證記憶體的回收。
如果需要空間動態縮小,可以考慮使用deque。如果非vector不可,可以用swap()來幫助你釋放記憶體。具體方法如下:
vector<int> nums; nums.push_back(1); nums.push_back(1); nums.push_back(2); nums.push_back(2); vector<int>().swap(nums); //或者nums.swap(vector<int> ())
或者如下所示,使用一對大括號,意思一樣的:
//加一對大括號是可以讓tmp退出{}的時候自動析構 { std::vector<int> tmp = nums; nums.swap(tmp); }
swap()是交換函式,使vector離開其自身的作用域,從而強制釋放vector所佔的記憶體空間,總而言之,釋放vector記憶體最簡單的方法是vector<int>.swap(nums)。當時如果nums是一個類的成員,不能把vector<int>.swap(nums)寫進類的解構函式中,否則會導致double free or corruption (fasttop)的錯誤,原因可能是重複釋放記憶體。標準解決方法如下:
template < class T > void ClearVector( vector< T >& vt ) { vector< T > vtTemp; veTemp.swap( vt ); }
3. 利用vector釋放指標
如果vector中存放的是指標,那麼當vector銷燬時,這些指標指向的物件不會被銷燬,那麼記憶體就不會被釋放。如下面這種情況,vector中的元素時由new操作動態申請出來的物件指標:
#include <vector> using namespace std; vector<void *> v;
每次new之後呼叫v.push_back()該指標,在程式退出或者根據需要,用以下程式碼進行記憶體的釋放:
for (vector<void *>::iterator it = v.begin(); it != v.end(); it ++) if (NULL != *it) { delete *it; *it = NULL; } v.clear();
當然還有種解決方法是重新定義一個函式,增加參數列示vector的開始和結束index。