From 31bcaeb11ec065d3ba0367a7f52b58983721564b Mon Sep 17 00:00:00 2001
From: wyy <wuyanyi09@gmail.com>
Date: Mon, 10 Feb 2014 11:08:26 +0800
Subject: [PATCH] fix bug: cmp function pair<string, uint> -> pair<string,
 double>

---
 src/KeywordExtractor.hpp | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/src/KeywordExtractor.hpp b/src/KeywordExtractor.hpp
index a78ea1f..d962502 100644
--- a/src/KeywordExtractor.hpp
+++ b/src/KeywordExtractor.hpp
@@ -138,8 +138,10 @@ namespace CppJieba
                     itr ++;
                 }
 
-                keywords.resize(MIN(topN, wordmap.size()));
-                partial_sort_copy(wordmap.begin(), wordmap.end(), keywords.begin(), keywords.end(), _cmp);
+                keywords.clear();
+                std::copy(wordmap.begin(), wordmap.end(), std::inserter(keywords, keywords.begin()));
+                topN = MIN(topN, keywords.size());
+                partial_sort(keywords.begin(), keywords.begin() + topN, keywords.end(), _cmp);
                 return true;
             }
         private:
@@ -153,7 +155,7 @@ namespace CppJieba
             }
 
         private:
-            static bool _cmp(const pair<string, uint>& lhs, const pair<string, uint>& rhs)
+            static bool _cmp(const pair<string, double>& lhs, const pair<string, double>& rhs)
             {
                 return lhs.second > rhs.second;
             }