solve 30

Author: Zhang Xiaodong

src/solution/mod.rs

@@ -25,3 +25,4 @@ mod s0026_remove_duplicates_from_sorted_array;
 mod s0027_remove_element;
 mod s0028_find_the_index_of_the_first_occurrence_in_a_string;
 mod s0029_divide_two_integers;
+mod s0030_substring_with_concatenation_of_all_words;

src/solution/s0030_substring_with_concatenation_of_all_words.rs

@@ -0,0 +1,121 @@
+/**
+ * [30] 串联所有单词的子串
+ *
+ * 给定一个字符串 s 和一个字符串数组 words。 words 中所有字符串 长度相同。
+ *  s 中的 串联子串 是指一个包含  words 中所有字符串以任意顺序排列连接起来的子串。
+ *
+ * 	例如，如果 words = ["ab","cd","ef"]， 那么 "abcdef"， "abefcd"，"cdabef"， "cdefab"，"efabcd"， 和 "efcdab" 都是串联子串。 "acdbef" 不是串联子串，因为他不是任何 words 排列的连接。
+ *
+ * 返回所有串联子串在 s 中的开始索引。你可以以 任意顺序 返回答案。
+ *  
+ * 示例 1：
+ *
+ * 输入：s = "barfoothefoobarman", words = ["foo","bar"]
+ * 输出：[0,9]
+ * 解释：因为 words.length == 2 同时 words[i].length == 3，连接的子字符串的长度必须为 6。
+ * 子串 "barfoo" 开始位置是 0。它是 words 中以 ["bar","foo"] 顺序排列的连接。
+ * 子串 "foobar" 开始位置是 9。它是 words 中以 ["foo","bar"] 顺序排列的连接。
+ * 输出顺序无关紧要。返回 [9,0] 也是可以的。
+ *
+ * 示例 2：
+ *
+ * 输入：s = "wordgoodgoodgoodbestword", words = ["word","good","best","word"]
+ * 输出：[]
+ * 解释：因为 words.length == 4 并且 words[i].length == 4，所以串联子串的长度必须为 16。
+ * s 中没有子串长度为 16 并且等于 words 的任何顺序排列的连接。
+ * 所以我们返回一个空数组。
+ *
+ * 示例 3：
+ *
+ * 输入：s = "barfoofoobarthefoobarman", words = ["bar","foo","the"]
+ * 输出：[6,9,12]
+ * 解释：因为 words.length == 3 并且 words[i].length == 3，所以串联子串的长度必须为 9。
+ * 子串 "foobarthe" 开始位置是 6。它是 words 中以 ["foo","bar","the"] 顺序排列的连接。
+ * 子串 "barthefoo" 开始位置是 9。它是 words 中以 ["bar","the","foo"] 顺序排列的连接。
+ * 子串 "thefoobar" 开始位置是 12。它是 words 中以 ["the","foo","bar"] 顺序排列的连接。
+ *  
+ * 提示：
+ *
+ * 	1 <= s.length <= 10^4
+ * 	1 <= words.length <= 5000
+ * 	1 <= words[i].length <= 30
+ * 	words[i] 和 s 由小写英文字母组成
+ *
+ */
+pub struct Solution {}
+
+// problem: https://leetcode.cn/problems/substring-with-concatenation-of-all-words/
+// discuss: https://leetcode.cn/problems/substring-with-concatenation-of-all-words/discuss/?currentPage=1&orderBy=most_votes&query=
+
+// submission codes start here
+
+impl Solution {
+    pub fn find_substring(s: String, words: Vec<String>) -> Vec<i32> {
+        let word_len = words[0].len();
+        let substr_len = word_len * words.len();
+        if substr_len > s.len() {
+            return vec![];
+        }
+        let mut sorted_words = words;
+        sorted_words.sort();
+
+        let mut res = vec![];
+        for i in 0..s.len() - substr_len {
+            let mut tmp_list = vec![];
+            let substr = &s[i..i + substr_len];
+            for j in (0..substr_len).step_by(word_len) {
+                tmp_list.push(&substr[j..j + word_len]);
+            }
+            tmp_list.sort();
+            if tmp_list == sorted_words {
+                res.push(i as i32);
+            }
+        }
+        res
+    }
+}
+
+// submission codes end
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_30() {
+        assert_eq!(
+            Solution::find_substring("hello".to_string(), vec!["ll".to_string()]),
+            vec![2],
+        );
+        assert_eq!(
+            Solution::find_substring(
+                "barfoothefoobarman".to_string(),
+                vec!["foo".to_string(), "bar".to_string()]
+            ),
+            vec![0, 9],
+        );
+        assert_eq!(
+            Solution::find_substring(
+                "wordgoodgoodgoodbestword".to_string(),
+                vec![
+                    "word".to_string(),
+                    "good".to_string(),
+                    "best".to_string(),
+                    "word".to_string()
+                ]
+            ),
+            vec![],
+        );
+        assert_eq!(
+            Solution::find_substring(
+                "barfoofoobarthefoobarman".to_string(),
+                vec![
+                    "bar".to_string(),
+                    "foo".to_string(),
+                    "the".to_string(),
+                ]
+            ),
+            vec![6, 9, 12],
+        );
+    }
+}