leetcode 140 单词拆分2 C++ 回溯

描述

给定一个非空字符串 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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

分析

按照普通的回溯方法可以实现，但是在遇到“aaaaaaaa…”超长字符串的时候会超时，所以在回溯的时候增加一个记忆机制
总的回溯思路是，backtrack(string s) 返回s当前开头可分割单词的之后的结果
比如"sanddog"的前半段是单词"cat"，当前回溯返回的是"sanddog"的分隔结果，所以将cat和返回的结果用空格连接，就可以加入到最终返回的res里了。
catsanddog – sanddog – dog --"“返回
cat sand dog – sand dog – dog --”"
那最后一步来说,s已经是""，自然返回""，此时的前半段词是dog，所以在dog这步返回的是dog + “”，依次向上
每次的返回结果用map也记录一下，这样可以直接返回。
比如不管是cats and dog还是cat sand dog，最终dog的部分就不需要重复递归来处理了，可以直接返回

代码

class Solution {
public:
	vector<string> wordBreak(string s, vector<string>& wordDict) {
		map<string, vector<string> > m;
		return helper(m, wordDict, s);
	}
	vector<string> helper(map<string, vector<string> >& m, vector<string>& wordDict, string s) {
		if (m.count(s)) return m[s];
		if (s.empty()) return { "" };
		vector<string> res;
		for (auto word : wordDict) {
			if (s.substr(0, word.size()) != word) continue;
			string temp = s.substr(word.size());
			vector<string> tmp = helper(m, wordDict, temp);
 			for (auto itm : tmp) {
				res.push_back(word + (itm.empty() ? "" : " " + itm));
			}
		}
		m[s] = res;
		return res;
	}
};