题目链接:leetcode.
想法很天真,哈希表存储某字母开头的所有word,回溯搜索是否能被拆分成功,然后超时
/*
35 / 45 个通过测试用例
状态:超出时间限制
"aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaab"
["a","aa","aaa","aaaa","aaaaa","aaaaaa","aaaaaaa","aaaaaaaa","aaaaaaaaa","aaaaaaaaaa"]
*/
class Solution {
unordered_map<char, vector<string>> M;
bool dfs(string s, int index)
{
if(index == s.size())
return true;
for(auto str : M[s[index]])
{
if(str == s.substr(index, str.size()))
{
if(dfs(s, index + str.size()))
return true;
}
}
return false;
}
public:
bool wordBreak(string s, vector<string>& wordDict) {
for(auto x : wordDict)
{
M[x[0]].push_back(x);
}
return dfs(s, 0);
}
};
题解表示应该使用动态规划哈,dp[i]
用来表示前i
个字符是否可拆分,dp[0]
表示空串默认为true
;从前往后更新dp[i]
,具体地,对于每一个长度i
,枚举长度j
(从1
到word表中最长的长度len
),当dp[i - j]
为真且substr(i - j, j)
在word表中存在时,dp[i]
才为真。
/*
执行用时:0 ms, 在所有 C++ 提交中击败了100.00%的用户
内存消耗:7.5 MB, 在所有 C++ 提交中击败了77.56%的用户
*/
class Solution {
public:
bool wordBreak(string s, vector<string>& wordDict) {
unordered_set<string> M;
int len = 0;//保存word的最长长度,以便剪枝
for(auto x : wordDict)
{
M.insert(x);
len = max(len, (int)x.size());
}
int N = s.size();
vector<bool> dp(N + 1, false);//保存前i个字符是否能成功拆分
dp[0] = true;//表示空串
for(int i = 1;i <= N;++i)
{
for(int j = 1;j <= len && i - j >= 0;++j)
{
dp[i] = dp[i] || (( j == 0 || dp[i - j]) && M.count(s.substr(i - j, j)));
}
}
return dp[N];
}
};