tuned the speed of analyzing

bab2min · Oct 30, 2023 · 8698de8 · 8698de8
1 parent a4a85c4
commit 8698de8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/Kiwi.cpp b/src/Kiwi.cpp
@@ -613,7 +613,7 @@ namespace kiwi
 
 		if (ret.empty())
 		{
-			const size_t n = pathes.size();
+			const size_t n = min(pathes.size(), topN * 2);
 			ret.resize(n);
 			spStatesByRet.resize(n);
 			parentMap.resize(n);

diff --git a/src/PathEvaluator.hpp b/src/PathEvaluator.hpp
@@ -1081,6 +1081,7 @@ namespace kiwi
 		{
 			return a.score > b.score;
 		});
+		if (ret.size() > topN * 2) ret.erase(ret.begin() + topN * 2, ret.end());
 		return ret;
 	}
 }