[Leetcode] Word Break、Word BreakII

最新推荐文章于 2021-01-02 23:29:20 发布

原创最新推荐文章于 2021-01-02 23:29:20 发布 · 7.8k 阅读

2 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

74 篇文章

订阅专栏

本文深入探讨了代码分割算法WordBreak及其在句子构造问题上的应用，详细阐述了解决字符串分割与句子构建的策略与实现细节。通过实例演示了如何使用动态规划解决字符串分割问题，并介绍了回溯算法在句子构造中的运用。文章旨在提供一种高效的方法来解决此类问题，对于编程爱好者和开发者具有较高的实用价值。

Word Break

Given a string s and a dictionary of words dict, determine if s can be segmented into a space-separated sequence of one or more dictionary words.

For example, given
s = "leetcode",
dict = ["leet", "code"].

Return true because "leetcode" can be segmented as "leet code".

才发现我一直把题意理解错了，原来是源串分出来的词只要都在字典里就好了，比如leetleet就可以分成两个leet都在词典里，原来不是所有字典里的词都要出现。

那就容易些了，另dp[i] 表示源串的前i个字符可以满足分割，那么 dp[ j ] 满足分割的条件是存在k 使得 dp [k] && substr[k,j]在字典里。

这是O(n2)的算法，注意下标，因为很恶心。

class Solution {
public:
    bool wordBreak(string s, unordered_set<string> &dict) {
        // Note: The Solution object is instantiated only once and is reused by each test case.
        int n = (int)s.size();
		vector<bool> dp(n + 1, false);
		dp[0]=true;
		for(int i=1;i<=n;i++)
		{
			if(dp[i-1])
			{
				int idx=i-1;
				for(int j=idx;j<n;j++)
				{
					string cur=s.substr(idx,j-idx+1);
					if(dict.count(cur)>0)
						dp[j+1]=true;
				}
			}
		}
		return dp[n];
	}
};

Word Break II

Given a string s and a dictionary of words dict, add spaces in s to construct a sentence where each word is a valid dictionary word.

Return all such possible sentences.

For example, given
s = "catsanddog",
dict = ["cat", "cats", "and", "sand", "dog"].

A solution is ["cats and dog", "cat sand dog"].

跟第一题一样，就是要返回所有可能的切分，做切分反应是用回溯，但是不剪枝肯定要超时。

这里用了一个math[i][j] 来表示 i--j 这一段是否可以切分，然后在dfs的时候利用看最后剩余的子串能否切分来剪枝。

class Solution {
public:
	vector<string> wordBreak(string s, unordered_set<string> &dict) 
	{
		int n=s.length();
		vector<vector<bool> > match(n+1,vector<bool>(n+1,false));
		for(int i=0;i<=n;i++)
			match[0][i]=true;
		for(int len=1;len<=n;len++)
		{
			for(int start=0;start+len<=n;start++)
			{
				string tmp=s.substr(start,len);
				if(dict.count(tmp)>0)
					match[len][start]=true;
				else
				{
					for(int left=1;left<len;left++)
					{
						match[len][start]=match[left][start]&&match[len-left][start+left];
						if(match[len][start])
							break;
					}
				}
			}
		}
		if(match[n][0]==false)
			return vector<string>();
		vector<string> ans;
		vector<string> had;
		dfs(s,0,match,had,ans,dict);
		return ans;
	}
	void dfs(string& s,int k,vector<vector<bool> >& match,vector<string>& had,vector<string>& ans,unordered_set<string> &dict)
	{
		int n=s.length();
		if(k>=n)
		{
			if(!had.empty())
			{
				string ret;
				for(int i=0;i<had.size();i++)
				{
					ret.append(had[i]);
					if(i!=had.size()-1)
						ret.push_back(' ');
				}
				ans.push_back(ret);
				return;
			}
		}
		for(int len=1;k+len<=n;len++)
		{
			string tmp=s.substr(k,len);
			if(dict.count(tmp)>0 && match[n-k-len][k+len])
			{
				had.push_back(tmp);
				dfs(s,k+len,match,had,ans,dict);
				had.pop_back();
			}
		}
	}	
};