·题目描述
给定两个字符串 s
和 p
,找到 s
中所有 p
的 异位词 的子串,返回这些子串的起始索引。不考虑答案输出的顺序。
异位词 指由相同字母重排列形成的字符串(包括相同的字符串)。
示例 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" 的异位词。
·解题思路
利用一个目标子串大小的窗口滑动,每次对滑动窗口中包含的字符串计数,若是与目标相同,则返回下标。
技巧在于,利用一个长度为26的数组来存储计数。
·Java代码
import java.util.ArrayList;
import java.util.List;
class Solution {
public static boolean isEqual(int[] a, int[] b ) {
for(int i =0 ; i < 26; i ++){
if(a[i]!= b[i]) return false;
}
return true;
}
public List<Integer> findAnagrams(String s, String p) {
int n = s.length(), m = p.length();;
List<Integer> res = new ArrayList<>();
int[] scount = new int[26], pcount = new int[26];
for(int i =0; i < m ; i ++){
pcount[p.charAt(i) - 'a'] ++;
}
for(int i = 0; i < n ; i ++){
scount[s.charAt(i) - 'a'] ++;
if(i >= m){
scount[s.charAt(i - m) - 'a'] --;
}
if (isEqual(pcount, scount)) res.add(i - m + 1);
}
return res;
}
}