[leetcode] 30. 串联所有单词的子串
题目
给定一个字符串 s
和一个字符串数组 words
。 words
中所有字符串 长度相同。
s
中的 串联子串 是指一个包含 words
中所有字符串以任意顺序排列连接起来的子串。
- 例如,如果
words = ["ab","cd","ef"]
, 那么"abcdef"
,"abefcd"
,"cdabef"
,"cdefab"
,"efabcd"
, 和"efcdab"
都是串联子串。"acdbef"
不是串联子串,因为他不是任何words
排列的连接。
返回所有串联子串在 s
中的开始索引。你可以以 任意顺序 返回答案。
示例 1:
输入:s = "barfoothefoobarman", words = ["foo","bar"]
输出:[0,9]
解释:因为 words.length == 2 同时 words[i].length == 3,连接的子字符串的长度必须为 6。
子串 "barfoo" 开始位置是 0。它是 words 中以 ["bar","foo"] 顺序排列的连接。
子串 "foobar" 开始位置是 9。它是 words 中以 ["foo","bar"] 顺序排列的连接。
输出顺序无关紧要。返回 [9,0] 也是可以的。
示例 2:
输入:s = "wordgoodgoodgoodbestword", words = ["word","good","best","word"]
输出:[]
解释:因为 words.length == 4 并且 words[i].length == 4,所以串联子串的长度必须为 16。
s 中没有子串长度为 16 并且等于 words 的任何顺序排列的连接。
所以我们返回一个空数组。
示例 3:
输入:s = "barfoofoobarthefoobarman", words = ["bar","foo","the"]
输出:[6,9,12]
解释:因为 words.length == 3 并且 words[i].length == 3,所以串联子串的长度必须为 9。
子串 "foobarthe" 开始位置是 6。它是 words 中以 ["foo","bar","the"] 顺序排列的连接。
子串 "barthefoo" 开始位置是 9。它是 words 中以 ["bar","the","foo"] 顺序排列的连接。
子串 "thefoobar" 开始位置是 12。它是 words 中以 ["the","foo","bar"] 顺序排列的连接。
提示:
1 <= s.length <= 104
1 <= words.length <= 5000
1 <= words[i].length <= 30
words[i]
和s
由小写英文字母组成
代码
#include<bits/stdc++.h>
using namespace std;
class Solution {
public:
vector<int> ans;
map<string,int> words2idx;
vector<int> findSubstring(string s, vector<string>& words) {
for (int i = 0; i < words.size(); i++) words2idx.insert(pair<string, int>(words[i], i));
int word_len = words[0].length(), seq_len = words.size();
vector<vector<int>> dp(word_len + 10, vector<int>(seq_len + 10, 0));
for (int i = 0; i < word_len; i++){
for (int j = 0; j < seq_len; j++){
dp[i][words2idx[words[j]]]++;
}
}
vector<string> subs(word_len + 10);
vector<int> cnt(word_len + 10, seq_len);
for (int i = 0; i < s.length(); i++){
for (int j = 0; j < word_len; j++){
if (i + j >= s.length()) continue;
subs[j] += s[i + j];
if (subs[j].length() % word_len == 0){
auto iter = words2idx.find(subs[j].substr(subs[j].length() - word_len));
if (iter != words2idx.end()){
if (dp[j][iter->second] > 0) cnt[j]-=1;
else cnt[j] -= 1e4;
dp[j][iter->second]--;
}
}
if (subs[j].length() == word_len * seq_len){
if (cnt[j] == 0){
ans.push_back(i + j - word_len * seq_len + 1);
}
string tmp = subs[j].substr(0, word_len);
auto iter = words2idx.find(tmp);
if (iter != words2idx.end()){
if (dp[j][iter->second] >= 0) cnt[j]+=1;
else cnt[j] += 1e4;
dp[j][iter->second]++;
}
subs[j].erase(0, word_len);
}
}
}
return ans;
}
};
int main(){
int n;
string s;
cin>>n;
cin>>s;
vector<string> words(n);
for (int i = 0; i < n; i++) cin>>words[i];
vector<int> ans = Solution().findSubstring(s, words);
for (int i = 0; i < ans.size(); i++) cout<<ans[i]<<" ";
}
滑动窗口。
其实可以不用开 words[i].length
个数组,可以选择不同的起点,然后一次在末尾增加 words[i].length
个字符,在头部删去 words[i].length
字符。相对而言更好理解。
本文来自博客园,作者:Un-Defined,转载请保留本文署名Un-Defined,并在文章顶部注明原文链接:https://www.cnblogs.com/EIPsilly/p/17735551.html