76 lines
2.0 KiB
Java
76 lines
2.0 KiB
Java
//给你一个二元数组 nums ，和一个整数 goal ，请你统计并返回有多少个和为 goal 的 非空 子数组。
|
||
//
|
||
// 子数组 是数组的一段连续部分。
|
||
//
|
||
//
|
||
//
|
||
// 示例 1：
|
||
//
|
||
//
|
||
//输入：nums = [1,0,1,0,1], goal = 2
|
||
//输出：4
|
||
//解释：
|
||
//有 4 个满足题目要求的子数组：[1,0,1]、[1,0,1,0]、[0,1,0,1]、[1,0,1]
|
||
//
|
||
//
|
||
// 示例 2：
|
||
//
|
||
//
|
||
//输入：nums = [0,0,0,0,0], goal = 0
|
||
//输出：15
|
||
//
|
||
//
|
||
//
|
||
//
|
||
// 提示：
|
||
//
|
||
//
|
||
// 1 <= nums.length <= 3 * 104
|
||
// nums[i] 不是 0 就是 1
|
||
// 0 <= goal <= nums.length
|
||
//
|
||
// Related Topics 数组 哈希表 前缀和 滑动窗口
|
||
// 👍 157 👎 0
|
||
|
||
package leetcode.editor.cn;
|
||
|
||
import java.util.HashMap;
|
||
import java.util.Map;
|
||
|
||
//930:和相同的二元子数组
|
||
public class BinarySubarraysWithSum {
|
||
public static void main(String[] args) {
|
||
//测试代码
|
||
Solution solution = new BinarySubarraysWithSum().new Solution();
|
||
}
|
||
|
||
//力扣代码
|
||
//leetcode submit region begin(Prohibit modification and deletion)
|
||
class Solution {
|
||
public int numSubarraysWithSum(int[] nums, int goal) {
|
||
//// 执行耗时:21 ms,击败了60.31% 的Java用户
|
||
//// 内存消耗:43.1 MB,击败了11.31% 的Java用户
|
||
// int sum = 0, count = 0;
|
||
// Map<Integer, Integer> map = new HashMap<>();
|
||
// for (int num : nums) {
|
||
// map.put(sum, map.getOrDefault(sum, 0) + 1);
|
||
// sum += num;
|
||
// count += map.getOrDefault(sum - goal, 0);
|
||
// }
|
||
// return count;
|
||
|
||
//// 执行耗时:2 ms,击败了98.67% 的Java用户
|
||
//// 内存消耗:41.7 MB,击败了76.94% 的Java用户
|
||
int sum = 0, count = 0;
|
||
int[] sums = new int[nums.length + 1];
|
||
for (int num : nums) {
|
||
sums[sum]++;
|
||
sum += num;
|
||
count += sum - goal >= 0 ? sums[sum - goal] : 0;
|
||
}
|
||
return count;
|
||
}
|
||
}
|
||
//leetcode submit region end(Prohibit modification and deletion)
|
||
|
||
} |