LeetCode HOT 100 —— 438. 找到字符串中所有字母异位词

题目

给定两个字符串 s 和 p，找到 s 中所有 p 的 异位词 的子串，返回这些子串的起始索引。不考虑答案输出的顺序。

异位词 指由相同字母重排列形成的字符串（包括相同的字符串）。

思路

双指针技巧：滑动窗口

滑动窗口的思想很简单，就是维护一个窗口，然后不断滑动，更新答案，算法大致逻辑如下：

int left = 0,right = 0;
while(right < s.length()){
	//向左增大窗口
	window.add(s[right]);
	right++;
	
	while(window needs shrink){
		//缩小窗口
		window.remove(s[left]);
		left++;
	}
}

这个算法技巧的时间复杂度是 O(N)，比字符串暴力算法要高效得多。

其实困扰的不是算法的思路，而是各种细节问题。比如说如何向窗口中添加新元素，如何缩小窗口，在窗口滑动的哪个阶段更新结果。

滑动窗口算法的代码框架：

/* 滑动窗口算法框架 */
void slidingWindow(string s, string p) {
	// 用 need 存储目标值p中各个单词出现的次数
    Map<Character, Integer> need = new HashMap<>();
    // 用window存储滑动窗口中有效字符出现的次数
    Map<Character, Integer> window = new HashMap<>();
   
    for (Character c : p.toCharArray()) //用 need 存储目标字符串p中各个单词出现的次数
    	need.put(c, need.getOrDefault(c, 0) + 1);
    
    int left = 0, right = 0;
    int valid = 0; 
    while (right < s.length()) {
        // c 是将移入窗口的字符
        char c = s[right];
        // 右移窗口
        right++;
        // 进行窗口内数据的一系列更新
        
        ...
        
        // 判断左侧窗口是否要收缩
        while (window needs shrink) {
            // d 是将移出窗口的字符
            char d = s[left];
            // 左移窗口
            left++;
            // 进行窗口内数据的一系列更新
            ...
        }
    }
}

套用这个框架，本题的题解如下：

class Solution {
        public List<Integer> findAnagrams(String s, String p) {

        Map<Character, Integer> window = new HashMap<>();//用window 存储滑动窗口中有效字符出现的次数
        Map<Character, Integer> need = new HashMap<>();// 用 need 存储目标值中各个单词出现的次数
        for (char c : p.toCharArray()) {
            need.put(c, need.getOrDefault(c, 0) + 1);
        }
        int left = 0;//  左指针
        int right = 0;// 右指针
        int valid = 0;//有效位，表示有效字符的个数，即当一个字符在need中和window的的出现频率相同时，为一个有效字符，此时有效位valid + 1
        List<Integer> result = new ArrayList<>();// 用于返回字母异位词的起始索引
        while (right < s.length()) {
            char c = s.charAt(right);//c是即将移入窗口的字符
            right++;
            if (need.containsKey(c)) {//如果这个字符在目标值need中出现
                window.put(c, window.getOrDefault(c, 0) + 1);//则加入滑动窗口，并更新出现频次
                if (window.get(c).equals(need.get(c))) {//如果出现频率相同，则有效位+1，注意这里用“==”有的测试用例会报错，因为Integer会缓存[-128，127]的数据，在这个区间用==和equals没区别，但是超过这个区间就必须要用equals了，因为Integer会创建两个新对象了
                    valid++;
                }
            }
            // 判断左侧窗⼝是否要收缩
            while (right - left >= p.length()) {//若当前滑动窗口大小大于目标值，则必定匹配失败，注意当前滑动窗口大小不是right - left + 1，而是right - left，因为right在前面就已经++过了
                if (valid == need.size()) {//如果有效位等于need大小，说明匹配成功
                    result.add(left);//将起始索引下标加入结果集中
                }
                char d = s.charAt(left);//即将移出窗口的字符
                left++;
                if (need.containsKey(d)) {//如果目标值need中包含这个字符
                    if (window.get(d).equals(need.get(d))) {//判断是不是有效字符，是的话有效位-1
                        valid--;
                    }
                    window.put(d, window.get(d) - 1);//同时将窗口中该字符的出现频次-1
                }
            }
        }
        return result;
    }
}