题意:
给定一个字符串 s 和一些长度相同的单词 words。
找出 s 中恰好可以由 words 中所有单词串联形成的子串的起始位置。
注意子串要与 words 中的单词完全匹配,中间不能有其他字符,但不需要考虑 words 中单词串联的顺序。
数据范围:
题目没给.
解法:
尺取,每次移动一个单词的长度即可.
因为每次只移动一个单词的长度,
设p为单词长度,那么尺取的起点只有%p=[0,p-1]种.
尺取部分的复杂度是O(p*n/p)=O(n)的.
字符串判定如果直接判每次是O(p)的,如果用字符串哈希就是O(1).
code:
class Solution {
public:
vector<int> findSubstring(string s, vector<string>& w) {
int n=s.size(),m=w.size(),p=w[0].size();
map<string,int>is;
for(auto& i:w){
is[i]++;
}
vector<int>ans;
for(int k=0;k<p;k++){
int sum=0;
int cnt=0;
map<string,int>mp;
for(int i=k;i+p-1<n;i+=p){
string temp=s.substr(i,p);
if(!is.count(temp)){
mp.clear();
sum=0;
cnt=0;
continue;
}
if(++mp[temp]==is[temp])sum++;
cnt++;
if(cnt>m){
string temp1=s.substr(i-m*p,p);
if(--mp[temp1]==is[temp1]-1)sum--;
cnt--;
}
if(cnt==m&&sum==is.size()){
ans.push_back(i-m*p+p);
}
}
}
return ans;
}
};