moved more stuff

1 files changed, 4 insertions, 7 deletions

diff --git a/string/kmp.cpp b/string/kmp.cpp
index 12ae3eb..421479e 100644
--- a/string/kmp.cpp
+++ b/string/kmp.cpp
@@ -1,18 +1,15 @@
 vector<int> kmpPreprocessing(const string& sub) {
 	vector<int> b(sz(sub) + 1);
 	b[0] = -1;
-	int i = 0, j = -1;
-	while (i < sz(sub)) {
+	for (int i = 0, j = -1; i < sz(sub);) {
 		while (j >= 0 && sub[i] != sub[j]) j = b[j];
-		i++; j++;
-		b[i] = j;
+		b[++i] = ++j;
 	}
 	return b;
 }
 vector<int> kmpSearch(const string& s, const string& sub) {
-	vector<int> pre = kmpPreprocessing(sub), result;
-	int i = 0, j = 0;
-	while (i < sz(s)) {
+	vector<int> result, pre = kmpPreprocessing(sub);
+	for (int i = 0, j = 0; i < sz(s);) {
 		while (j >= 0 && s[i] != sub[j]) j = pre[j];
 		i++; j++;
 		if (j == sz(sub)) {