LeetCode140:单词拆分ii

直接在leetcode上做的,牛客上把词典wordDict改成了哈希表,懒得改了,牛客上的就没写。

思路:

table[i][j]代表从字符串s的i到j位置([i,j])是否能为一个词典中的单词,为1表示可以,为0表示不行。

e.g.

s:pineapplepenapple
wordDict = ["apple", "pen", "applepen", "pine", "pineapple"]

首先初始化table表,即初始化table[0][x],在词典中找能令0~x成为字典中词的单词。

这里找到pine与pineapple,即初始化table[0][3]与table[0][8]为1。

初始化完成后开始建表

第一遍遍历i=0,即字符串从0开始,找到table[0][3]:pine与table[0][8]:pineapple

遍历过程中,根据遍历结果继续建表,比如找到table[0][3]后相当于肯定0~3位置是一个符合要求的单词了。

现在以3~s.size()-1为一个新串,在里面继续找单词,如现在找到table[0][3]了,则继续在applepenapple中找,与初始化

时一样,继续在词典wordDict中找,如现在找到apple,则令table[4][8]为1,如此循环下去直到建表完成。

测试样例中有个样例aaaaaaaaaaaa.....,一直超时,这是由于没有判断样例是否能拆分就开始拆分造成的结果。

因此先判断字符串能否拆分,判断很简单,只要遍历table[i][s.size()-1]是否全为0即可,如果全为0,说明无法拆分到s字符串的最后一个字母,应该返回空。

返回结果很简单,根据表,首先遍历table[0][i],找到table[0][3]=1了,则跳到table[4],在table[4][i]中找1。。。。直到table[j][i]中j=s.size(),则将结果输入到result。

总体来说

table[i][j]代表从i到j能否构成一个词,则我们只要把表格建好,表格代表了所有单词可能出现的情况。

然后从0开始,找到第一个单词,然后以第一个单词后面的串继续找下一个单词,直到最后。

class Solution {
public:
    vector<string> wordBreak(string s, vector<string>& wordDict) {
        vector<vector<int>> table;
		table.resize(s.size());
		for(int i=0;i<table.size();i++)
		{
			table[i].resize(s.size());
		}
		//初始化:
		for(int i=0;i<wordDict.size();i++)
		{
			if(strstr(s.data(),wordDict[i].data())-s.data()==0)
			{
				int idx=wordDict[i].size();
				table[0][idx-1]=1;
			}
		}
		//建表
		for(int i=0;i<table.size()-1;i++)
		{
			for(int j=0;j<table[i].size();j++)
			{
				if(table[i][j]==1)
				{
					if(j==table[i].size()-1)
						continue;
					string tmp=s.substr(j+1);
					for(int k=0;k<wordDict.size();k++)
					{
						if(strstr(tmp.data(),wordDict[k].data())-tmp.data()==0)
						{
							int idx=wordDict[k].size();
							table[j+1][j+idx]=1;
						}
					}
				}
			}
		}
		vector<string> result;
		bool flag=false;
		for(int i=0;i<table.size();i++)
		{
			if(table[i][s.size()-1]==1)
				flag=true;
		}
		if(!flag)
			return result;
		//查表返回结果
		for(int i=0;i<s.size();i++)
		{
			if(table[0][i]==1)
			{
				int idx=i+1;
				string tmp=s.substr(0,idx);
				fillCore(result,table,i+1,tmp,s,s.size());
			}
		}
		return result;
    }
private:
	void fillCore(vector<string>& result,vector<vector<int>>& table,int start,string s,string& input,int length)
	{
		if(start==length)
			result.push_back(s);
		for(int i=start;i<length;i++)
		{
			if(table[start][i]==1)//pineapplepenapple
			{
				string tmp=s;
				s=s+" "+input.substr(start,i-start+1);
				fillCore(result,table,i+1,s,input,length);
				s=tmp;
			}
		}
	}
};

  

原文地址:https://www.cnblogs.com/lxy-xf/p/11132871.html