word_break追加

Author: calloc134

src/main.rs

@@ -1,5 +1,9 @@
-mod is_subsequence;
+mod word_break;
+mod word_break_rev;
 fn main() {
-    let result = is_subsequence::Solution::is_subsequence("abc".to_string(), "ahbgdc".to_string());
+    let result = word_break::Solution::word_break(
+        "leetcode".to_string(),
+        vec!["leet".to_string(), "code".to_string()],
+    );
     println!("{:?}", result);
 }

src/word_break.rs

@@ -0,0 +1,108 @@
+pub struct Solution;
+impl Solution {
+    pub fn word_break(s: String, word_dict: Vec<String>) -> bool {
+        let mut dp = vec![false; s.len() + 1];
+        dp[0] = true;
+        for i in 1..=s.len() {
+            for j in 0..i {
+                if dp[j] && word_dict.contains(&s[j..i].to_string()) {
+                    dp[i] = true;
+                    break;
+                }
+            }
+        }
+        dp[s.len()]
+    }
+
+    pub fn word_break_rev(s: String, word_dict: Vec<String>) -> bool {
+        let mut dp = vec![false; s.len() + 1];
+        dp[0] = true;
+        for i in 1..=s.len() {
+            for j in (0..i).rev() {
+                if dp[j] && word_dict.contains(&s[j..i].to_string()) {
+                    dp[i] = true;
+                    break;
+                }
+            }
+        }
+        dp[s.len()]
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_1() {
+        assert_eq!(
+            Solution::word_break(
+                "leetcode".to_string(),
+                vec!["leet".to_string(), "code".to_string()]
+            ),
+            true
+        );
+    }
+
+    #[test]
+    fn test_2() {
+        assert_eq!(
+            Solution::word_break(
+                "applepenapple".to_string(),
+                vec!["apple".to_string(), "pen".to_string()]
+            ),
+            true
+        );
+    }
+
+    #[test]
+    fn test_3() {
+        assert_eq!(
+            Solution::word_break(
+                "catsandog".to_string(),
+                vec![
+                    "cats".to_string(),
+                    "dog".to_string(),
+                    "sand".to_string(),
+                    "and".to_string(),
+                    "cat".to_string()
+                ]
+            ),
+            false
+        );
+    }
+
+    #[test]
+    fn test_4() {
+        assert_eq!(
+            Solution::word_break(
+                "cars".to_string(),
+                vec!["car".to_string(), "ca".to_string(), "rs".to_string()]
+            ),
+            true
+        );
+    }
+
+    #[test]
+    fn test_5() {
+        assert_eq!(
+            Solution::word_break(
+                "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaab".to_string(),
+                // vec!["a","aa","aaa","aaaa","aaaaa","aaaaaa","aaaaaaa","aaaaaaaa","aaaaaaaaa","aaaaaaaaaa"]
+                vec![
+                    "a".to_string(),
+                    "aa".to_string(),
+                    "aaa".to_string(),
+                    "aaaa".to_string(),
+                    "aaaaa".to_string(),
+                    "aaaaaa".to_string(),
+                    "aaaaaaa".to_string(),
+                    "aaaaaaaa".to_string(),
+                    "aaaaaaaaa".to_string(),
+                    "aaaaaaaaaa".to_string()
+                ]
+            ),
+            false
+        );
+    }
+}

src/word_break_rev.rs

@@ -0,0 +1,96 @@
+pub struct Solution;
+impl Solution {
+    pub fn word_break(s: String, word_dict: Vec<String>) -> bool {
+        let mut dp = vec![false; s.len() + 1];
+        dp[0] = true;
+
+        for i in 1..=s.len() {
+            for j in (0..i).rev() {
+                if dp[j] && word_dict.contains(&s[j..i].to_string()) {
+                    dp[i] = true;
+                    break;
+                }
+            }
+        }
+
+        dp[s.len()]
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_1() {
+        assert_eq!(
+            Solution::word_break(
+                "leetcode".to_string(),
+                vec!["leet".to_string(), "code".to_string()]
+            ),
+            true
+        );
+    }
+
+    #[test]
+    fn test_2() {
+        assert_eq!(
+            Solution::word_break(
+                "applepenapple".to_string(),
+                vec!["apple".to_string(), "pen".to_string()]
+            ),
+            true
+        );
+    }
+
+    #[test]
+    fn test_3() {
+        assert_eq!(
+            Solution::word_break(
+                "catsandog".to_string(),
+                vec![
+                    "cats".to_string(),
+                    "dog".to_string(),
+                    "sand".to_string(),
+                    "and".to_string(),
+                    "cat".to_string()
+                ]
+            ),
+            false
+        );
+    }
+
+    #[test]
+    fn test_4() {
+        assert_eq!(
+            Solution::word_break(
+                "cars".to_string(),
+                vec!["car".to_string(), "ca".to_string(), "rs".to_string()]
+            ),
+            true
+        );
+    }
+
+    #[test]
+    fn test_5() {
+        assert_eq!(
+            Solution::word_break(
+                "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaab".to_string(),
+                // vec!["a","aa","aaa","aaaa","aaaaa","aaaaaa","aaaaaaa","aaaaaaaa","aaaaaaaaa","aaaaaaaaaa"]
+                vec![
+                    "a".to_string(),
+                    "aa".to_string(),
+                    "aaa".to_string(),
+                    "aaaa".to_string(),
+                    "aaaaa".to_string(),
+                    "aaaaaa".to_string(),
+                    "aaaaaaa".to_string(),
+                    "aaaaaaaa".to_string(),
+                    "aaaaaaaaa".to_string(),
+                    "aaaaaaaaaa".to_string()
+                ]
+            ),
+            false
+        );
+    }
+}