suffix array find string

KHvic · KHvic · commit 3819334b4c05 · 2020-10-18T03:31:05.000+08:00
diff --git a/Suffix-Array/sa.cpp b/Suffix-Array/sa.cpp
@@ -39,4 +39,15 @@ struct SA {
             }
         }
     }
+    bool findString(const string& t, int spos, int epos) {
+        int lo = 0, hi = n-1;
+        int len = epos-spos+1;
+        while(lo<hi) {
+            int mid = (lo+hi)/2;
+            int res = s.compare(sa[mid], len, t, spos, len);
+            (res>=0) ? hi = mid : lo = mid+1;
+        }
+        return s.compare(sa[lo], len, t, spos, len)==0;
+    }
+    bool findString(const string& t) {return findString(t, 0, t.size()-1);}
 };

Original file line number	Diff line number	Diff line change
`@@ -39,4 +39,15 @@ struct SA {`
`39`	`39`	`}`
`40`	`40`	`}`
`41`	`41`	`}`
	`42`	`+ bool findString(const string& t, int spos, int epos) {`
	`43`	`+ int lo = 0, hi = n-1;`
	`44`	`+ int len = epos-spos+1;`
	`45`	`+ while(lo<hi) {`
	`46`	`+ int mid = (lo+hi)/2;`
	`47`	`+ int res = s.compare(sa[mid], len, t, spos, len);`
	`48`	`+ (res>=0) ? hi = mid : lo = mid+1;`
	`49`	`+ }`
	`50`	`+ return s.compare(sa[lo], len, t, spos, len)==0;`
	`51`	`+ }`
	`52`	`+ bool findString(const string& t) {return findString(t, 0, t.size()-1);}`
`42`	`53`	`};`