【LeetCode】 找到字符串中所有字母异位词Java题解

356 阅读1分钟

小知识,大挑战!本文正在参与“程序员必备小知识”创作活动。

题目描述

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

异位词 指字母相同,但排列不同的字符串。


示例 1:

输入: s = "cbaebabacd", p = "abc"
输出: [0,6]
解释:
起始索引等于 0 的子串是 "cba", 它是 "abc" 的异位词。
起始索引等于 6 的子串是 "bac", 它是 "abc" 的异位词。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/find-all-anagrams-in-a-string
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

思路分析

  • 今天的算法每日一题是求解异位词问题,异位词 指字母相同,但排列不同的字符串。
  • 我们解决异位词问题,一般使用 ASCII 码知识,通过数组存储字符出现的次数。
  • 根据题意,首先统计 p 中字符出现的次数,然后动态更新 s 中字符出现的次数,比较 cnt 数组,即可得到答案。

通过代码

class Solution {
    public List<Integer> findAnagrams(String s, String p) {
        List<Integer> ans = new ArrayList<>();
        int n = s.length();
        int m = p.length();
        if (n < m) {
            return ans;
        }

        int[] sCnt = new int[26];
        int[] pCnt = new int[26];
        for (int i = 0; i < m; i++) {
            sCnt[s.charAt(i) - 'a']++;
            pCnt[p.charAt(i) - 'a']++;
        }

        if (Arrays.equals(sCnt, pCnt)) {
            ans.add(0);
        }

        for (int i = m; i < n; i++) {
            sCnt[s.charAt(i) - 'a']++;
            sCnt[s.charAt(i - m) - 'a']--;
            if (Arrays.equals(sCnt, pCnt)) {
                ans.add(i - m + 1);
            }
        }
        return ans;
    }
}


image.png

总结

  • 上述算法的时间复杂度是O(n), 空间复杂度是O(n)
  • 坚持算法每日一题,加油!