题目:
给定一个字符串 s 和一个非空字符串 p,找到 s 中所有是 p 的字母异位词的子串,返回这些子串的起始索引。
字符串只包含小写英文字母,并且字符串 s 和 p 的长度都不超过 20100。
说明:
字母异位词指字母相同,但排列不同的字符串。
不考虑答案输出的顺序。
示例 1:
输入:
s: "cbaebabacd" p: "abc"
输出:
[0, 6]
解释:
起始索引等于 0 的子串是 "cba", 它是 "abc" 的字母异位词。
起始索引等于 6 的子串是 "bac", 它是 "abc" 的字母异位词。
示例 2:
输入:
s: "abab" p: "ab"
输出:
[0, 1, 2]
解释:
起始索引等于 0 的子串是 "ab", 它是 "ab" 的字母异位词。
起始索引等于 1 的子串是 "ba", 它是 "ab" 的字母异位词。
起始索引等于 2 的子串是 "ab", 它是 "ab" 的字母异位词。
题解:
参考labuladong的模板:
import java.util.*;
class Solution {
public List<Integer> findAnagrams(String s, String p) {
// 构建目标容器和滑窗命中容器
HashMap<Character, Integer> need = new HashMap<>();
HashMap<Character, Integer> window = new HashMap<>();
// 初始化目标容器
for (int i = 0; i < p.length(); i++) {
need.put(p.charAt(i), need.getOrDefault(p.charAt(i), 0) + 1);
}
// 设置滑窗参数
int left = 0;
int right = 0;
int valid_num = 0;
List<Integer> markIndex = new ArrayList();
// 滑窗到达边缘停止
while (right < s.length()) {
// 先扩大滑窗
char addChar = s.charAt(right);
right++;
// 如果元素是目标元素,算作滑窗命中
if (need.containsKey(addChar)) {
window.put(addChar, window.getOrDefault(addChar, 0) + 1);
if (window.get(addChar).equals(need.get(addChar))) {
valid_num++;
}
}
// 如果当前命中数包含所有目标元素,则开始缩减窗口
while (valid_num == need.size()) {
// 这题要求子串不包括其他字符,必须连续,所以要校验长度
// 如果长度符合,那就记住前下标
if (right - left == p.length()) {
markIndex.add(left);
}
char removeChar = s.charAt(left);
left++;
// 判断和缩小滑窗边界
if (need.containsKey(removeChar)) {
if (need.get(removeChar).equals(window.get(removeChar))) {
valid_num--;
}
window.put(removeChar, window.get(removeChar) - 1);
}
}
}
// 返回值
return markIndex;
}
}
public class Main {
public static void main(String[] args) {
String s = "ababab";
String p = "aba";
Solution ab = new Solution();
List<Integer> res = ab.findAnagrams(s, p);
System.out.println("结果:" + res);
}
}