一、题目介绍
给定一个非空字符串 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/word-break
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
二、解题思路
本题采用回溯的方法枚举所有情况,进而判断给定字符串是否可以被wordDict中的单词拆分。回溯方法计算一些例子会超时,因此改用动态规划的方法。
动态规划:
(1)状态:dp[i]代表字符串前i个字符,是否可以被wordDict中的单词拆分。
(2)初始状态:dp[0] = true;
(3)状态转移方程 dp[i] = dp[j] && wordDict.find(s.substr(j, i-j)) != wordDict.end(); (其中 0 < j < i )
三、解题代码
class Solution {
public:
bool wordBreak(string s, vector<string>& wordDict) {
if(s.empty() || wordDict.empty())
return false;
set<string> st(wordDict.begin(), wordDict.end());
int n = s.size();
vector<bool> dp(n+1, false); //字符串s的前i个字符能否拆分成wordDict中的单词
dp[0] = true; //初始状态
int maxlen = 0;
for(int i = 0; i < wordDict.size(); ++i)
{
maxlen = max(maxlen, (int)wordDict[i].size()); //计算字典中单词的最大长度
}
//状态转移
for(int i = 1; i <= n; ++i)
{
//优化 j不一定从0开始,因为单词的最大长度为 maxlen,所以其实位置为max(i-maxlen, 0)
for(int j = max(i-maxlen, 0); j < i; ++j)
{
if(dp[j] && st.count(s.substr(j, i-j)) == 1)
{
dp[i] = true;
break;
}
}
}
return dp[n];
}
};