力扣【438】找到字符串中所有字母异位词

题目:

给定一个字符串 s 和一个非空字符串 p,找到 s 中所有是 p 的字母异位词的子串,返回这些子串的起始索引。

字符串只包含小写英文字母,并且字符串 s 和 p 的长度都不超过 20100。

说明:

字母异位词指字母相同,但排列不同的字符串。
不考虑答案输出的顺序。
示例 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" 的字母异位词。

题解:

参考labuladong的模板:

import java.util.*;

class Solution {
    public List<Integer> findAnagrams(String s, String p) {
        // 构建目标容器和滑窗命中容器
        HashMap<Character, Integer> need = new HashMap<>();
        HashMap<Character, Integer> window = new HashMap<>();
        // 初始化目标容器
        for (int i = 0; i < p.length(); i++) {
            need.put(p.charAt(i), need.getOrDefault(p.charAt(i), 0) + 1);
        }
        // 设置滑窗参数
        int left = 0;
        int right = 0;
        int valid_num = 0;
        List<Integer> markIndex = new ArrayList();
        // 滑窗到达边缘停止
        while (right < s.length()) {
            // 先扩大滑窗
            char addChar = s.charAt(right);
            right++;
            // 如果元素是目标元素,算作滑窗命中
            if (need.containsKey(addChar)) {
                window.put(addChar, window.getOrDefault(addChar, 0) + 1);
                if (window.get(addChar).equals(need.get(addChar))) {
                    valid_num++;
                }
            }
            // 如果当前命中数包含所有目标元素,则开始缩减窗口
            while (valid_num == need.size()) {
                // 这题要求子串不包括其他字符,必须连续,所以要校验长度
                // 如果长度符合,那就记住前下标
                if (right - left == p.length()) {
                    markIndex.add(left);
                }
                char removeChar = s.charAt(left);
                left++;
                // 判断和缩小滑窗边界
                if (need.containsKey(removeChar)) {
                    if (need.get(removeChar).equals(window.get(removeChar))) {
                        valid_num--;
                    }
                    window.put(removeChar, window.get(removeChar) - 1);
                }
            }
        }
        // 返回值
        return markIndex;
    }
}

public class Main {
    public static void main(String[] args) {
        String s = "ababab";
        String p = "aba";
        Solution ab = new Solution();
        List<Integer> res = ab.findAnagrams(s, p);
        System.out.println("结果:" + res);
    }
}

猜你喜欢

转载自blog.csdn.net/qq1922631820/article/details/114011064