LeetCode热题100-11、12题

11、找到字符串中所有的字母异位词

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

// 双指针维护一段区间，如果这段区间的每个字母数出现和模板一样，即可
class Solution {
    public List<Integer> findAnagrams(String s, String p) {
        int sLen = s.length(),pLen = p.length();

        if (sLen < pLen) {
            return new ArrayList<Integer>();
        }

        List<Integer> ans = new ArrayList<Integer>();
        int[] sCount = new int[26];
        int[] pCount = new int[26];
        for (int i = 0;i < pLen;i++) {
            ++sCount[s.charAt(i) - 'a'];
            ++pCount[p.charAt(i) - 'a'];
        }

        if (Arrays.equals(sCount,pCount)) {
            ans.add(0);
        }

        for (int i = 0;i < sLen - pLen;++i) {
            --sCount[s.charAt(i) - 'a'];
            ++sCount[s.charAt(i + pLen) - 'a'];

            if (Arrays.equals(sCount,pCount)) {
                ans.add(i + 1);
            }
        }

        return ans;

    }
}

12、和为k的子数组

给你一个整数数组 nums 和一个整数 k ，请你统计并返回 该数组中和为 k 的子数组的个数 。

子数组是数组中元素的连续非空序列。
示例 1：

输入：nums = [1,1,1], k = 2
输出：2
示例 2：

输入：nums = [1,2,3], k = 3
输出：2

// 数学里面的公式：1、i,j之间的数组和为 a[i+1] + a[i + 2] + ... + a[j],即sum[j] - sum[i]
// 当枚举到j时，我们知道sum[j],想知道是否存在i到j和为k，只需要判断 sum[j] - k = sum[i]
// 特殊情况i = j时，sum[j] = sum[i],这是会发现i到j和为k，只需要判断是否存在0，故mp初始值为(0,1)
class Solution {
    public int subarraySum(int[] nums, int k) {
        int count = 0,pre = 0;
        HashMap<Integer,Integer> mp = new HashMap<>();

        mp.put(0,1);

        for (int i = 0;i < nums.length;i++) {
            pre += nums[i];
            if (mp.containsKey(pre - k)) {
                count += mp.get(pre - k);
            }
            mp.put(pre,mp.getOrDefault(pre,0) + 1);
        }
        return count;
    }
}