剑指Offer || 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/problems/VabMRr
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题解

参考：剑指Offer || 014.字符串中的变位词，开一个list对正确结果的初始下标进行存储即可。

注意对于diff的理解，不是有多少个不同的元素，是有多少种不同元素，后面变化时，对于不在p中的元素变化也会对diff产生影响。

代码：

class Solution {
    public List<Integer> findAnagrams(String s, String p) {
    	List<Integer> result=new ArrayList<>();
    	int n=s.length(),m=p.length();
    	if(n<m) return result;
    	int[] cnt=new int[26];
    	for(int i=0;i<m;i++) {
    		cnt[p.charAt(i)-'a']--;
    		cnt[s.charAt(i)-'a']++;
    	}
    	int diff=0;
    	for(int c:cnt) {
    		if(c!=0) diff++;
    	}
    	if(diff==0) result.add(0);
    	for(int i=m;i<n;i++) {
    		int x=s.charAt(i)-'a',y=s.charAt(i-m)-'a';
    		if(cnt[x]==0) diff++;
    		cnt[x]++;
    		if(cnt[x]==0) diff--;
    		if(cnt[y]==0) diff++;
    		cnt[y]--;
    		if(cnt[y]==0) diff--;
    		if(diff==0)
    			result.add(i-m+1);
    	}
    	return result;
    }
}