summaryrefslogtreecommitdiff
path: root/src/main/HashSubstring.java
diff options
context:
space:
mode:
Diffstat (limited to 'src/main/HashSubstring.java')
-rw-r--r--src/main/HashSubstring.java135
1 files changed, 135 insertions, 0 deletions
diff --git a/src/main/HashSubstring.java b/src/main/HashSubstring.java
new file mode 100644
index 0000000..baef360
--- /dev/null
+++ b/src/main/HashSubstring.java
@@ -0,0 +1,135 @@
+import java.io.*;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.StringTokenizer;
+import java.util.concurrent.ThreadLocalRandom;
+
+public class HashSubstring {
+
+ private static FastScanner in;
+ private static PrintWriter out;
+
+ public static void main(String[] args) throws IOException {
+ in = new FastScanner();
+ out = new PrintWriter(new BufferedOutputStream(System.out));
+ printOccurrences(rabinKarp(readInput()));
+ out.close();
+ }
+
+ private static Data readInput() throws IOException {
+ String pattern = in.next();
+ String text = in.next();
+ return new Data(pattern, text);
+ }
+
+ private static void printOccurrences(List<Integer> ans) {
+ for (Integer cur : ans) {
+ out.print(cur);
+ out.print(" ");
+ }
+ }
+
+ static List<Integer> getOccurrences(Data input) {
+ String s = input.pattern, t = input.text;
+ int m = s.length(), n = t.length();
+ List<Integer> occurrences = new ArrayList<>();
+ for (int i = 0; i + m <= n; ++i) {
+ boolean equal = true;
+ for (int j = 0; j < m; ++j) {
+ if (s.charAt(j) != t.charAt(i + j)) {
+ equal = false;
+ break;
+ }
+ }
+ if (equal)
+ occurrences.add(i);
+ }
+ return occurrences;
+ }
+
+ static class Data {
+ String pattern;
+ String text;
+
+ Data(String pattern, String text) {
+ this.pattern = pattern;
+ this.text = text;
+ }
+ }
+
+ static long polyHash(String S, int p, int x) {
+ long hash = 0;
+
+ for (int i = S.length() - 1; i >= 0; --i)
+ hash = (hash * x + S.charAt(i)) % p;
+
+ return hash;
+ }
+
+ static long[] preComputeHashes(String T, int pattenLength, int p, int x) {
+ long[] H = new long[T.length() - pattenLength + 1];
+ String S = T.substring(T.length() - pattenLength);
+ H[T.length() - pattenLength] = polyHash(S, p, x);
+ long y = 1;
+ for (int i = 0; i < pattenLength; i++) {
+ y = (y * x) % p;
+ }
+ for (int i = T.length() - pattenLength - 1; i >= 0; --i) {
+ H[i] = (x * H[i + 1] + T.charAt(i) - y * T.charAt(i + pattenLength)) % p;
+ }
+ return H;
+ }
+
+ static List<Integer> rabinKarp(Data input) {
+ String T = input.text;
+ String P = input.pattern;
+ int p = 1000000007;
+ int x = ThreadLocalRandom.current().nextInt(1, p);
+ List<Integer> positions = new ArrayList<>();
+ long pHash = polyHash(P, p, x);
+ long[] H = preComputeHashes(T, P.length(), p, x);
+ for (int i = 0; i < T.length() - P.length() + 1; i++) {
+ if (pHash != H[i]) {
+ continue;
+ }
+ if (T.substring(i, i + P.length()).equals(P))
+ positions.add(i);
+ }
+ return positions;
+ }
+
+ static List<Integer> rabinKarp(String P, String T) {
+ int p = 1000000007;
+ int x = ThreadLocalRandom.current().nextInt(1, p);
+ List<Integer> positions = new ArrayList<>();
+ long pHash = polyHash(P, p, x);
+ long[] H = preComputeHashes(T, P.length(), p, x);
+ for (int i = 0; i < T.length() - P.length() + 1; i++) {
+ if (pHash != H[i]) {
+ continue;
+ }
+ if (T.substring(i, i + P.length()).equals(P))
+ positions.add(i);
+ }
+ return positions;
+ }
+
+ static class FastScanner {
+ private BufferedReader reader;
+ private StringTokenizer tokenizer;
+
+ FastScanner() {
+ reader = new BufferedReader(new InputStreamReader(System.in));
+ tokenizer = null;
+ }
+
+ String next() throws IOException {
+ while (tokenizer == null || !tokenizer.hasMoreTokens()) {
+ tokenizer = new StringTokenizer(reader.readLine());
+ }
+ return tokenizer.nextToken();
+ }
+ }
+}
+
+