串联所有单词的子串
问题描述
给定一个字符串 s 和一些长度相同的单词 words。找出 s 中恰好可以由 words 中所有单词串联形成的子串的起始位置。
注意子串要与 words 中的单词完全匹配,中间不能有其他字符,但不需要考虑 words 中单词串联的顺序。
示例 1:
输入:
s = "barfoothefoobarman",
words = ["foo","bar"]
输出:0 9
解释:
从索引 0 和 9 开始的子串分别是 "barfoo" 和 "foobar" 。输出时,按照索引由小到大顺序输出。
示例 2:
输入:
s = "wordgoodgoodgoodbestword",
words = ["word","good","best","word"]
输出:-1
s中的子串无法由words串联得到,所以输出-1
可使用以下main函数:
int main()
{
string s,str;
vector<string> words;
int n;
cin>>s;
cin>>n;
for(int i=0; i<n; i++)
{
cin>>str;
words.push_back(str);
}
vector<int> res=Solution().findSubstring(s, words);
if (res.size()> 0)
for(int i=0; i<res.size(); i++)
{
if (i> 0)
cout<<" ";
cout<<res[i];
}
else
cout<<-1;
return 0;
}
输入说明
首先收入字符串s,
然后输入words中单词的数目n,
最后输入n个字符串,表示words中的单词
输出说明
按照索引由小到大顺序输出,或者输出-1.
输入范例
barfoothefoobarman
2
foo bar
输出范例
0 9
实现思路
思路与判断是否为字母异位词一样,只不过本题是判断单词是否异位。(但本代码对那个巨长的a字符串会运行超时)
实现代码
#include <iostream>#include<sstream>#include <vector>#include<math.h>#include<unordered_map>using namespace std;class Solution {
public:bool isAnagram(vector<string>&a,vector<string>&b){unordered_map<string,int>mp;for(int i = 0;i<a.size();i++){mp[a[i]]++;}for(int i = 0;i<b.size();i++){mp[b[i]]--;}for(std::unordered_map<string,int>::iterator it = mp.begin();it!=mp.end();++it){if(it->second!=0) return false;}return true;}vector<int> findSubstring(string s, vector<string>& words) {vector<string> str;vector<int>res;/*for(int i = 0;i<words.size();i++){str += words[i];}*/int n = words.size()*words[0].size();//words所有单词的长度for(int i = 0;i<s.size()-n+1;i++){string tem = "";for(int j = i;j<i+n;j++){tem += s[j];if(tem.size()==words[0].size()){str.push_back(tem);tem = "";}}if(isAnagram(words,str)) res.push_back(i);str.clear();//要记得清空上一次获得的字符串}return res;}
};int main(){string s,str;vector<string> words;int n;cin>>s;cin>>n;for(int i=0; i<n; i++){cin>>str;words.push_back(str);}vector<int> res=Solution().findSubstring(s, words);if (res.size()> 0)for(int i=0; i<res.size(); i++){if (i> 0)cout<<" ";cout<<res[i];}elsecout<<-1;return 0;}