add shrink for vector in DictTrie.hpp

yanyiwu · Apr 11, 2014 · bb6c3f9 · bb6c3f9
1 parent 0ca598b
commit bb6c3f9
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/DictTrie.hpp b/src/DictTrie.hpp
@@ -80,6 +80,7 @@ namespace CppJieba
             {
                 assert(!_getInitFlag());
                 _loadDict(filePath, _nodeInfos);
+                _shrink(_nodeInfos);
                 _freqSum = _calculateFreqSum(_nodeInfos);
                 assert(_freqSum);
                 _minLogFreq = _calculateLogFreqAndGetMinValue(_nodeInfos, _freqSum);
@@ -114,6 +115,7 @@ namespace CppJieba
                     words.push_back(dictUnits[i].word);
                     valuePointers.push_back(&dictUnits[i]);
                 }
+
                 TrieType * trie = new TrieType(words, valuePointers);
                 return trie;
             }
@@ -172,6 +174,11 @@ namespace CppJieba
                 return minLogFreq;
             }
 
+            void _shrink(vector<DictUnit>& units) const
+            {
+                vector<DictUnit>(units.begin(), units.end()).swap(units);
+            }
+
 
     };
 }