forked from mengli/leetcode
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathSubstring with Concatenation of All Words.java
More file actions
59 lines (55 loc) · 1.71 KB
/
Substring with Concatenation of All Words.java
File metadata and controls
59 lines (55 loc) · 1.71 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
You are given a string, S, and a list of words, L, that are all of the same length. Find all starting indices of substring(s) in S that is a concatenation of each word in L exactly once and without any intervening characters.
For example, given:
S: "barfoothefoobarman"
L: ["foo", "bar"]
You should return the indices: [0,9].
(order does not matter).
public class Solution {
public ArrayList<Integer> findSubstring(String S, String[] L) {
int slen = S.length();
int m = L.length;
ArrayList<Integer> ret = new ArrayList<Integer>();
if (m == 0) return ret;
int n = L[0].length();
HashMap<String, Integer> smap = new HashMap<String, Integer>();
for (String s : L) {
if (smap.containsKey(s)) {
smap.put(s, smap.get(s).intValue() + 1);
} else {
smap.put(s, 1);
}
}
if (m * n > slen) return ret;
int start = 0, end = m * n - 1;
int index = 0;
while (end < slen) {
index = check(S, start, end, m, n, L, smap);
if (index >= 0) ret.add(index);
start++;
end++;
}
return ret;
}
private int check(String s, int start, int end, int m, int n, String[] L, HashMap<String, Integer> smap) {
HashMap<String, Integer> exitMap = new HashMap<String, Integer>();
int sidx = start;
while(sidx <= end) {
String k = s.substring(sidx, sidx + n);
if (smap.containsKey(k)) {
if (exitMap.containsKey(k)) {
Integer i = exitMap.get(k);
if (i == smap.get(k)) return -1;
else {
exitMap.put(k, i + 1);
}
} else {
exitMap.put(k, 1);
}
sidx += n;
} else {
return -1;
}
}
return start;
}
}