1. 程式人生 > >Leetcode: Substring with Concatenation of All Words

Leetcode: Substring with Concatenation of All Words

You are given a string, S, and a list of words, L, that are all of the same length. Find all starting indices of substring(s) in S that is a concatenation of each word in L exactly once and without any intervening characters.

For example, given:
S"barfoothefoobarman"
L["foo", "bar"]

You should return the indices: [0,9]

.
(order does not matter).

一開始理解錯題意了,以為只要有L中的單詞在S中沒有干擾詞彙的連續出現就要記下來呢,提交發現不對,人家是each word in L exactly once.是要所有L中的詞都出現一遍,中間不能有其他的干擾詞彙。並且L中的詞也會有重複。

錯誤程式碼:

vector<int> findSubstring(string S, vector<string> &L) {
        // Note: The Solution object is instantiated only once.
		set<string> dic;
		for(int i = 0; i < L.size(); i++)
			dic.insert(L[i]);

		vector<int> res;
		int wordlen = L[0].size();
		set<string>::iterator it;
		int i = 0;
		bool isLastOk = false;
		while(i < S.size())
		{
			string sub = S.substr(i,wordlen);
			it = dic.find(sub);
			if(it != dic.end())
			{
				if(!isLastOk)
				{
					res.push_back(i);
					isLastOk = true;
				}
					i += wordlen;
			}else
			{
				isLastOk = false;
				i++;
			}
		}
		return res;
    }

正確程式碼如下:
vector<int> findSubstring(string S, vector<string> &L) {
        // Note: The Solution object is instantiated only once.
		map<string,int> words;
		map<string,int> cur;
		int wordNum = L.size();
		for(int i = 0; i < wordNum; i++)
			words[L[i]]++;
		int wordLen = L[0].size();
		vector<int> res;
		//if(S.size() < wordLen*wordNum)return res;
		for(int i = 0; i <= (int)S.size()-wordLen*wordNum; i++)
		{
			cur.clear();
			int j;
			for(j = 0; j < wordNum; j++)
			{
				string word = S.substr(i+j*wordLen, wordLen);
				if(words.find(word) == words.end())
					break;
				cur[word]++;
				if(cur[word]>words[word])
					break;
			}
			if(j == wordNum)
				res.push_back(i);
		}
		return res;
	}