|
| 1 | +# 题目描述(困难难度) |
| 2 | + |
| 3 | + |
| 4 | + |
| 5 | +给定一个字符串 s ,给定 n 个单词 word,找出所有子串的开始下标,使得子串包含了给定的所有单词,顺序可以不对应。如果有重复的单词,比如有 [ " foo " , " foo " ] 那么子串也必须含有两个 " foo ",也就是说个数必须相同。 |
| 6 | + |
| 7 | +# 解法一 |
| 8 | + |
| 9 | +参考 leetCode 里的 [solution](https://leetcode.com/problems/substring-with-concatenation-of-all-words/discuss/13658/Easy-Two-Map-Solution-(C%2B%2BJava)) |
| 10 | + |
| 11 | +首先,最直接的思路,判断每个子串是否符合,符合就把下标保存起来,最后返回即可。 |
| 12 | + |
| 13 | + |
| 14 | + |
| 15 | +如上图,利用循环变量 i ,依次后移,判断每个子串是否符合即可。 |
| 16 | + |
| 17 | +怎么判断子串是否符合?这也是这个题的难点了,由于子串包含的单词顺序并不需要固定,如果是两个单词 A,B,我们只需要判断子串是否是 AB 或者 BA 即可。如果是三个单词 A,B,C 也还好,只需要判断子串是否是 ABC,或者 ACB,BAC,BCA,CAB,CBA 就可以了,但如果更多单词呢?那就崩溃了。 |
| 18 | + |
| 19 | +[链接](https://leetcode.com/problems/substring-with-concatenation-of-all-words/discuss/13658/Easy-Two-Map-Solution-(C%2B%2BJava))的作者提出了,用两个 HashMap 来解决。首先,我们把所有的单词存到 HashMap 里,key 直接存单词,value 存单词出现的个数(因为给出的单词可能会有重复的,所以可能是 1 或 2 或者其他)。然后扫描子串的单词,如果当前扫描的单词在之前的 HashMap 中,就把该单词存到新的 HashMap 中,并判断新的 HashMap 中该单词的 value 是不是大于之前的 HashMap 该单词的 value ,如果大了,就代表该子串不是我们要找的,接着判断下一个子串就可以了。如果不大于,那么我们接着判断下一个单词的情况。子串扫描结束,如果子串的全部单词都符合,那么该子串就是我们找的其中一个。看下具体的例子。 |
| 20 | + |
| 21 | +看下图,我们把 words 存到一个 HashMap 中。 |
| 22 | + |
| 23 | + |
| 24 | + |
| 25 | +然后遍历子串的每个单词。 |
| 26 | + |
| 27 | + |
| 28 | + |
| 29 | +第一个单词在 HashMap1 中,然后我们把 foo 存到 HashMap2 中。并且比较此时 foo 的 value 和 HashMap1 中 foo 的 value,1 < 2,所以我们继续扫描。 |
| 30 | + |
| 31 | + |
| 32 | + |
| 33 | +第二个单词也在 HashMap1 中,然后把 foo 存到 HashMap2 中,因为之前已经存过了,所以更新它的 value 为 2 ,然后继续比较此时 foo 的 value 和 HashMap1 中 foo 的 value,2 <= 2,所以继续扫描下一个单词。 |
| 34 | + |
| 35 | + |
| 36 | + |
| 37 | +第三个单词也在 HashMap1 中,然后把 foo 存到 HashMap2 中,因为之前已经存过了,所以更新它的 value 为 3,然后继续比较此时 foo 的 value 和 HashMap1 中 foo 的 value,3 > 2,所以表明该字符串不符合。然后判断下个子串就好了。 |
| 38 | + |
| 39 | +当然上边的情况都是单词在 HashMap1 中,如果不在的话就更好说了,不在就表明当前子串肯定不符合了,直接判断下个子串就好了。 |
| 40 | + |
| 41 | +看一下代码吧 |
| 42 | + |
| 43 | +```java |
| 44 | +public List<Integer> findSubstring(String s, String[] words) { |
| 45 | + List<Integer> res = new ArrayList<Integer>(); |
| 46 | + int wordNum = words.length; |
| 47 | + if (wordNum == 0) { |
| 48 | + return res; |
| 49 | + } |
| 50 | + int wordLen = words[0].length(); |
| 51 | + //HashMap1 存所有单词 |
| 52 | + HashMap<String, Integer> allWords = new HashMap<String, Integer>(); |
| 53 | + for (String w : words) { |
| 54 | + int value = allWords.getOrDefault(w, 0); |
| 55 | + allWords.put(w, value + 1); |
| 56 | + } |
| 57 | + //遍历所有子串 |
| 58 | + for (int i = 0; i < s.length() - wordNum * wordLen + 1; i++) { |
| 59 | + //HashMap2 存当前扫描的字符串含有的单词 |
| 60 | + HashMap<String, Integer> hasWords = new HashMap<String, Integer>(); |
| 61 | + int num = 0; |
| 62 | + //判断该子串是否符合 |
| 63 | + while (num < wordNum) { |
| 64 | + String word = s.substring(i + num * wordLen, i + (num + 1) * wordLen); |
| 65 | + //判断该单词在 HashMap1 中 |
| 66 | + if (allWords.containsKey(word)) { |
| 67 | + int value = hasWords.getOrDefault(word, 0); |
| 68 | + hasWords.put(word, value + 1); |
| 69 | + //判断当前单词的 value 和 HashMap1 中该单词的 value |
| 70 | + if (hasWords.get(word) > allWords.get(word)) { |
| 71 | + break; |
| 72 | + } |
| 73 | + } else { |
| 74 | + break; |
| 75 | + } |
| 76 | + num++; |
| 77 | + } |
| 78 | + //判断是不是所有的单词都符合条件 |
| 79 | + if (num == wordNum) { |
| 80 | + res.add(i); |
| 81 | + } |
| 82 | + } |
| 83 | + return res; |
| 84 | +} |
| 85 | +``` |
| 86 | + |
| 87 | +时间复杂度:假设 s 的长度是 n,words 里有 m 个单词,那么时间复杂度就是 O(n * m)。 |
| 88 | + |
| 89 | +空间复杂度:两个 HashMap,假设 words 里有 m 个单词,就是 O(m)。 |
| 90 | + |
| 91 | +# 解法二 |
| 92 | + |
| 93 | +参考 https://leetcode.com/problems/substring-with-concatenation-of-all-words/discuss/13656/An-O(N)-solution-with-detailed-explanation。 |
| 94 | + |
| 95 | +我们在解法一中,每次移动一个字符。 |
| 96 | + |
| 97 | + |
| 98 | + |
| 99 | +现在为了方便讨论,我们每次移动一个单词的长度,也就是 3 个字符,这样所有的移动被分成了三类。 |
| 100 | + |
| 101 | + |
| 102 | + |
| 103 | + |
| 104 | + |
| 105 | + |
| 106 | + |
| 107 | +以上三类我们以第一类从 0 开始移动为例,讲一下如何对算法进行优化,有三种需要优化的情况。 |
| 108 | + |
| 109 | +* 情况一:当子串完全匹配,移动到下一个子串的时候。 |
| 110 | + |
| 111 | +  |
| 112 | + |
| 113 | + 在解法一中,对于 i = 3 的子串,我们肯定是从第一个 foo 开始判断。但其实前两个 foo 都不用判断了 ,因为在判断上一个 i = 0 的子串的时候我们已经判断过了。所以解法一中的 HashMap2 每次并不需要清空从 0 开始,而是可以只移除之前 i = 0 子串的第一个单词 bar 即可,然后直接从箭头所指的 foo 开始就可以了。 |
| 114 | + |
| 115 | +* 情况二:当判断过程中,出现不符合的单词。 |
| 116 | + |
| 117 | +  |
| 118 | + |
| 119 | + 但判断 i = 0 的子串的时候,出现了 the ,并不在所给的单词中。所以此时 i = 3,i = 6 的子串,我们其实并不需要判断了。我们直接判断 i = 9 的情况就可以了。 |
| 120 | + |
| 121 | +* 情况三:判断过程中,出现的是符合的单词,但是次数超了。 |
| 122 | + |
| 123 | +  |
| 124 | + |
| 125 | + 对于 i = 0 的子串,此时判断的 bar 其实是在 words 中的,但是之前已经出现了一次 bar,所以 i = 0 的子串是不符合要求的。此时我们只需要往后移动窗口,i = 3 的子串将 foo 移除,此时子串中一定还是有两个 bar,所以该子串也一定不符合。接着往后移动,当之前的 bar 被移除后,此时 i = 6 的子串,就可以接着按正常的方法判断了。 |
| 126 | + |
| 127 | + 所以对于出现 i = 0 的子串的情况,我们可以直接从 HashMap2 中依次移除单词,当移除了之前次数超的单词的时候,我们就可以正常判断了,直接从移除了超出了次数的单词后,也就是 i = 6 开始判断就可以了。 |
| 128 | + |
| 129 | + 看一下代码吧。 |
| 130 | + |
| 131 | + ```java |
| 132 | + public List<Integer> findSubstring(String s, String[] words) { |
| 133 | + List<Integer> res = new ArrayList<Integer>(); |
| 134 | + int wordNum = words.length; |
| 135 | + if (wordNum == 0) { |
| 136 | + return res; |
| 137 | + } |
| 138 | + int wordLen = words[0].length(); |
| 139 | + HashMap<String, Integer> allWords = new HashMap<String, Integer>(); |
| 140 | + for (String w : words) { |
| 141 | + int value = allWords.getOrDefault(w, 0); |
| 142 | + allWords.put(w, value + 1); |
| 143 | + } |
| 144 | + //将所有移动分成 wordLen 类情况 |
| 145 | + for (int j = 0; j < wordLen; j++) { |
| 146 | + HashMap<String, Integer> hasWords = new HashMap<String, Integer>(); |
| 147 | + int num = 0; //记录当前 HashMap2(这里的 hasWords 变量)中有多少个单词 |
| 148 | + //每次移动一个单词长度 |
| 149 | + for (int i = j; i < s.length() - wordNum * wordLen + 1; i = i + wordLen) { |
| 150 | + boolean hasRemoved = false; //防止情况三移除后,情况一继续移除 |
| 151 | + while (num < wordNum) { |
| 152 | + String word = s.substring(i + num * wordLen, i + (num + 1) * wordLen); |
| 153 | + if (allWords.containsKey(word)) { |
| 154 | + int value = hasWords.getOrDefault(word, 0); |
| 155 | + hasWords.put(word, value + 1); |
| 156 | + //出现情况三,遇到了符合的单词,但是次数超了 |
| 157 | + if (hasWords.get(word) > allWords.get(word)) { |
| 158 | + // hasWords.put(word, value); |
| 159 | + hasRemoved = true; |
| 160 | + int removeNum = 0; |
| 161 | + //一直移除单词,直到次数符合了 |
| 162 | + while (hasWords.get(word) > allWords.get(word)) { |
| 163 | + String firstWord = s.substring(i + removeNum * wordLen, i + (removeNum + 1) * wordLen); |
| 164 | + int v = hasWords.get(firstWord); |
| 165 | + hasWords.put(firstWord, v - 1); |
| 166 | + removeNum++; |
| 167 | + } |
| 168 | + num = num - removeNum + 1; //加 1 是因为我们把当前单词加入到了 HashMap 2 中 |
| 169 | + i = i + (removeNum - 1) * wordLen; //这里依旧是考虑到了最外层的 for 循环,看情况二的解释 |
| 170 | + break; |
| 171 | + } |
| 172 | + //出现情况二,遇到了不匹配的单词,直接将 i 移动到该单词的后边(但其实这里 |
| 173 | + //只是移动到了出现问题单词的地方,因为最外层有 for 循环, i 还会移动一个单词 |
| 174 | + //然后刚好就移动到了单词后边) |
| 175 | + } else { |
| 176 | + hasWords.clear(); |
| 177 | + i = i + num * wordLen; |
| 178 | + num = 0; |
| 179 | + break; |
| 180 | + } |
| 181 | + num++; |
| 182 | + } |
| 183 | + if (num == wordNum) { |
| 184 | + res.add(i); |
| 185 | + |
| 186 | + } |
| 187 | + //出现情况一,子串完全匹配,我们将上一个子串的第一个单词从 HashMap2 中移除 |
| 188 | + if (num > 0 && !hasRemoved) { |
| 189 | + String firstWord = s.substring(i, i + wordLen); |
| 190 | + int v = hasWords.get(firstWord); |
| 191 | + hasWords.put(firstWord, v - 1); |
| 192 | + num = num - 1; |
| 193 | + } |
| 194 | + |
| 195 | + } |
| 196 | + |
| 197 | + } |
| 198 | + return res; |
| 199 | + } |
| 200 | + |
| 201 | + ``` |
| 202 | + |
| 203 | + 时间复杂度:算法中外层的两个for 循环的次数肯定是所有的子串,假设是 n。考虑一下,最极端的情况,每个子串的判断都进了 while 循环,wordNum 等于 m。对于解法一,因为每次都是从头判断,所以 while 循环循环了 m 次。但这里我们由于没有清空,所以每次只判断新加入的单词就可以了,只需判断一次,所以时间复杂度是 O(n)。 |
| 204 | + |
| 205 | + 或者换一种理解方式,判断子串是否符合,本质上也就是判断每个单词符不符合,假设 s 的长度是 n,那么就会大约有 n 个子串,也就是会有 n 个单词。而对于每个单词,我们只有刚开始判断符不符合的时候访问一次,还有就是把它移除的时候访问一次,所以每个单词最多访问 2 次,所以时间复杂度是 O(n)。 |
| 206 | + |
| 207 | + 空间复杂度:没有变化,依旧是两个 HashMap, 假设 words 里有 m 个单词,就是 O(m)。 |
| 208 | + |
| 209 | + # 总 |
| 210 | + |
| 211 | + 这道题最大的亮点就是应用了 HashMap 了吧,使得我们不再纠结于子串包含单词的顺序。然后对于算法的优化上,还是老思路,去分析哪些判断是不必要的,然后把它除之。 |
0 commit comments