剑指 Offer II 015. 字符串中的所有变位词
给定两个字符串 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” 的变位词。
提示:
1 <= s.length, p.length <= 3 * 104
s 和 p 仅包含小写字母
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/VabMRr
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
分析
在剑指 Offer II 015 的基础上修改即可。
题解(Java)
class Solution {
public List<Integer> findAnagrams(String s, String p) {
List<Integer> ans = new ArrayList<>();
if (s.length() < p.length()) return ans;
int[] counts = new int[26];
for (int i = 0; i < p.length(); i++) {
counts[p.charAt(i) - 'a']++;
counts[s.charAt(i) - 'a']--;
}
if (judge(counts)) ans.add(0);
for (int i = p.length(); i < s.length(); i++) {
counts[s.charAt(i) - 'a']--;
counts[s.charAt(i - p.length()) - 'a']++;
if (judge(counts)) ans.add(i - p.length() + 1);
}
return ans;
}
public boolean judge(int[] counts) {
for (int count : counts) {
if (count != 0) return false;
}
return true;
}
}