前缀和的一些题目-CSDN博客

和为k的数组

public int subarraySum(int[] nums, int k) {
     int pre = 0;
     int count = 0;
     //一个哈希表存储当前的前缀和
     HashMap<Integer,Integer> mp = new HashMap<>();
     mp.put(0,1);
     for (int i = 0; i < nums.length; i++) {
         //当前的前缀和
         pre += nums[i];
         //如果之前存在已需要的哈希和等于pre-k，则说明存在和为k的子数组
         if (mp.containsKey(pre - k)){
             count += mp.get(pre - k);
         }
         mp.put(pre, mp.getOrDefault(pre,0)+1);
     }
     return count;
 }

除自身以外数组的乘积

使用一个pre[]数组来记录前缀和，pre[i]表示位置i之前所有的0~i-1的乘积之和
使用一个post[]数组来记录后缀和，post[i]表示位置i之后的所有i~len-1的乘积之和
因此除了元素nums[i]的乘积之和为pre[i] * post[i]

public static int[] productExceptSelf(int[] nums) {
	int len = nums.length;
    int[] pre = new int[len];
    pre[0] = 1;

    int[] post = new int[len];
    post[len-1] = 1;

    for (int i = 1; i < len; i++) {
        pre[i] = nums[i-1] * pre[i-1];
    }

    for (int i = len-2; i >= 0; i--) {
        post[i] = nums[i+1] * post[i+1];
    }

    int[] ans = new int[len];
    for (int i = 0; i < len; i++) {
        ans[i] = pre[i] * post[i];
    }
    return ans;
 }