fixed bab2min/kiwipiepy#168

bab2min · May 1, 2024 · e506efb · e506efb
1 parent 37bfa60
commit e506efb
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/Kiwi.cpp b/src/Kiwi.cpp
@@ -898,16 +898,18 @@ namespace kiwi
 	{
 		ret.clear();
 		auto* cur = first;
-		for (size_t i = 0; i < nodes.size() && cur != last; ++i)
+		for (size_t i = 0; i < nodes.size(); ++i)
 		{
+			while (cur != last && nodes[i].startPos >= cur->end) ++cur;
+			if (cur == last) break;
+
 			if (cur->begin <= nodes[i].startPos && nodes[i].endPos <= cur->end)
 			{
 				ret.emplace_back(cur - first);
 			}
 			else
 			{
 				ret.emplace_back(-1);
-				if (nodes[i].startPos >= cur->end) ++cur;
 			}
 		}
 		ret.resize(nodes.size(), -1);