LeetCode 139. 单词拆分

题目描述

给定一个非空字符串 s 和一个包含非空单词列表的字典 wordDict，判定 s 是否可以被空格拆分为一个或多个在字典中出现的单词。

说明：

• 拆分时可以重复使用字典中的单词。
• 你可以假设字典中没有重复的单词。

示例 1：

输入: s = "leetcode", wordDict = ["leet", "code"]
输出: true
解释: 返回 true 因为 "leetcode" 可以被拆分成 "leet code"。

示例 2：

输入: s = "applepenapple", wordDict = ["apple", "pen"]
输出: true
解释: 返回 true 因为 "applepenapple" 可以被拆分成 "apple pen apple"。
     注意你可以重复使用字典中的单词。

示例 3：

输入: s = "catsandog", wordDict = ["cats", "dog", "sand", "and", "cat"]
输出: false

问题分析

此题使用DFS来做，但我一开始进行直接了当的DFS，结果超时了。后来参考别人的思路发现要用一个哈希表<string, bool> memo来存储字符串能否被拆分，这样如果后面在递归判断某子串s能否被拆分时，如果这个子串s在之前求出并存入了哈希表memo，就可以直接取出结果并返回。将字符串s传入DFS函数开始递归，先在哈希表memo中找，如果找到了直接返回已有结果。如果字符串为空，那么返回true。然后定义一个变量res，初始值为false。接下来开始for循环，在数组wordDict中遍历字符串，当遇到某一个字符串与s中开头部分匹配的话，把字符串s中除了开头剩下的部分传入DFS函数中进行判断，如果DFS函数返回true，则说明字符串s是可以被拆分的，将变量res置为true。在for循环结束后，如果s可拆分，则此时res为true，否则为false。所以我们将字符串s与相应的结果res记录在哈希表memo中，并返回。

代码实现

class Solution {
public:
    bool wordBreak(string s, vector<string>& wordDict) {
        unordered_map<string, bool> memo;
        return DFS(s, wordDict, memo);
    }
    
    bool DFS(string s, vector<string>& wordDict, unordered_map<string, bool>& memo){
        if(memo.count(s))
            return memo[s];
        if(s.empty())
            return true;
        bool res = false;
        for(string str : wordDict){
            if(s.substr(0, str.size()) == str){
                if(DFS(s.substr(str.size()), wordDict, memo))
                    res = true;      
            }
        }
        memo[s] = res;
        return memo[s];
    }
};