题目
给定两个字符串 s 和 p,找到 s 中所有 p 的 异位词 的子串,返回这些子串的起始索引。不考虑答案输出的顺序。
异位词 指由相同字母重排列形成的字符串(包括相同的字符串)。
示例
输入: s = “cbaebabacd”, p = “abc”
输出: [0,6]
解释:
起始索引等于 0 的子串是 “cba”, 它是 “abc” 的异位词。
起始索引等于 6 的子串是 “bac”, 它是 “abc” 的异位词。
输入: s = “abab”, p = “ab”
输出: [0,1,2]
解释:
起始索引等于 0 的子串是 “ab”, 它是 “ab” 的异位词。
起始索引等于 1 的子串是 “ba”, 它是 “ab” 的异位词。
起始索引等于 2 的子串是 “ab”, 它是 “ab” 的异位词。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/find-all-anagrams-in-a-string
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
方法1:滑动窗口+双指针
参数:
1、初始化两个窗口:need、window
2、初始化两个指针:left、right
3、初始化满足要求的字符个数:valid
每次右指针移动一下,就要判断+修改window窗口,判断左侧窗口是否需要收缩。
如果左侧窗口需要收缩,每次收缩都要判断修改window窗口。
Java实现1
class Solution {
public List<Integer> findAnagrams(String s, String p) {
//需求窗口、滑动窗口
Map<Character, Integer> need = new HashMap<>();
Map<Character, Integer> window = new HashMap<>();
//存索引
List<Integer> res = new ArrayList<>();
//需求窗口遍历记录
for (char c : p.toCharArray()) {
need.put(c, need.getOrDefault(c, 0) + 1);
}
//左右指针、valid表示窗口中满足 need 条件的字符个数
int left = 0, right = 0, valid = 0;
while (right < s.length()){
//注意right要 ++
char c = s.charAt(right);
right++;
//更新window窗口的数据
if (need.containsKey(c)) {
window.put(c, window.getOrDefault(c, 0) + 1);
if (window.get(c).equals(need.get(c))) valid++;
}
//判断左侧窗口是否要收缩
while (right - left >= p.length()){
//窗口符合条件,添加起始索引
if (valid == need.size()) res.add(left);
//注意left要++
char d = s.charAt(left);
left++;
//进行窗口数据更新
if (need.containsKey(d)) {
if (need.get(d).equals(window.get(d))) valid--;
window.put(d, window.get(d) - 1);
}
}
}
return res;
}
}
Java实现2
class Solution {
public List<Integer> findAnagrams(String s, String p) {
int[] needs = new int[26];
int[] window = new int[26];
List<Integer> res = new ArrayList<>();
if (p.length() > s.length()) return res;
// 设置窗口大小
int windowSize = p.length();
for (int i = 0; i < windowSize; i++) {
needs[p.charAt(i) - 'a']++;
window[s.charAt(i) - 'a']++;
}
// 滑窗
for (int i = 0; i <= s.length() - windowSize; i++) {
if (Arrays.equals(needs, window)) {
res.add(i);
}
// 停止滑窗
if (i == s.length() - windowSize) break;
window[s.charAt(i + windowSize) - 'a']++;
window[s.charAt(i) - 'a']--;
}
return res;
}
}