LeetCode-找到字符串中所有字母异位词(滑动窗口)

深夜无眠T2024-08-19 0:32

题目描述

给定两个字符串 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" 的异位词。

思路解析

本题可以使用滑动窗口来求解，先使用两个数组来维护窗口内每个词出现的频率，一个数组记录要求解的字符串中单词的频率，一个维护当前窗口内单词的频率，通过不断地移动窗口时将两个数组进行对比，如果相同则证明此时是异位词，记录下当前索引。

代码如下：

java 复制代码

class Solution {
    public List<Integer> findAnagrams(String s, String p) {
        List<Integer> ret = new ArrayList<>();
        if (s == null || p == null || s.length() < p.length()) return ret;
        int[] sCount = new int[26];
        int[] pCount = new int[26];
        for (int i = 0; i < p.length(); i++) {
            sCount[s.charAt(i) - 'a']++;
            pCount[p.charAt(i) - 'a']++;
        }
        if (Arrays.equals(sCount, pCount)) {
            ret.add(0);
        }
        int left = 0;
        int right = left + p.length() - 1;

        while (true) { 
            sCount[s.charAt(left) - 'a']--;
            left++;
            right++;
            if (right >= s.length()) break;
            sCount[s.charAt(right) - 'a']++;
            if (Arrays.equals(sCount, pCount)) {
                ret.add(left);
            }
        }
        return ret;
    }
}