一、题目介绍
给定一个非空字符串 s 和一个包含非空单词列表的字典 wordDict,在字符串中增加空格来构建一个句子,使得句子中所有的单词都在词典中。返回所有这些可能的句子。
说明:
分隔时可以重复使用字典中的单词。
你可以假设字典中没有重复的单词。
示例 1:
输入:
s = "catsanddog"
wordDict = ["cat", "cats", "and", "sand", "dog"]
输出:
[
"cats and dog",
"cat sand dog"
]
示例 2:
输入:
s = "pineapplepenapple"
wordDict = ["apple", "pen", "applepen", "pine", "pineapple"]
输出:
[
"pine apple pen apple",
"pineapple pen apple",
"pine applepen apple"
]
解释: 注意你可以重复使用字典中的单词。
示例 3:
输入:
s = "catsandog"
wordDict = ["cats", "dog", "sand", "and", "cat"]
输出:
[]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/word-break-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
二、解题思路
(1)通过动态规划的方法,记录前i个字符串是否可以用词典中的单词拆分,即确定状态。
(2)通过深度优先搜索的方法,遍历所有情况的组合。
三、解题代码
class Solution {
public:
int n;
vector<string> res;
vector<bool> dp; //代表前i个字符是否可以被词典中的单词拆分
vector<string> wordBreak(string s, vector<string>& wordDict) {
n = s.size();
dp.assign(n+1, false);
dp[0] = true; //初始状态
set<string> st(wordDict.begin(), wordDict.end());
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)
{
for(int j = max(0, i-maxlen); j < i; ++j)
{
if(dp[j] && st.count(s.substr(j, i-j)) == 1)
{
dp[i] = true;
break;
}
}
}
if(dp[n])
{
deque<string> que;
dfs(res, n, wordDict, s, que, st);
}
return res;
}
//深度优先搜索所有情况
void dfs(vector<string>& res, int nsize, vector<string>& wordDict, string s, deque<string>& que, set<string>& st)
{
if(nsize == 0)
{
string stemp;
for(auto item : que)
{
stemp += item + " ";
}
if(!stemp.empty())
{
res.push_back(string(stemp.begin(), stemp.end()-1));
}
return;
}
for(int i = 0; i < nsize; ++i)
{
if(dp[i]) //首先前i个字符能够被词典中的单词拆分
{
string str = s.substr(i, nsize-i); //先找到最后一个可以拆分的单词
if(st.count(str) == 1) //如果剩下的字符组成的单词在词典中可以查到
{
que.push_front(str);
dfs(res, i, wordDict, s, que, st);
que.pop_front(); //恢复现场
}
}
}
}
};