930:和相同的二元子数组

This commit is contained in:
huangge1199 2021-07-08 15:59:54 +08:00
parent 71d8add7c6
commit 03f4e33ca0
4 changed files with 109 additions and 3 deletions

View File

@ -0,0 +1,76 @@
//给你一个二元数组 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)
}

View File

@ -0,0 +1,32 @@
<p>给你一个二元数组 <code>nums</code> ,和一个整数 <code>goal</code> ,请你统计并返回有多少个和为 <code>goal</code><strong> 非空</strong> 子数组。</p>
<p><strong>子数组</strong> 是数组的一段连续部分。</p>
<p> </p>
<p><strong>示例 1</strong></p>
<pre>
<strong>输入:</strong>nums = [1,0,1,0,1], goal = 2
<strong>输出:</strong>4
<strong>解释:</strong>
有 4 个满足题目要求的子数组:[1,0,1]、[1,0,1,0]、[0,1,0,1]、[1,0,1]
</pre>
<p><strong>示例 2</strong></p>
<pre>
<strong>输入:</strong>nums = [0,0,0,0,0], goal = 0
<strong>输出:</strong>15
</pre>
<p> </p>
<p><strong>提示:</strong></p>
<ul>
<li><code>1 <= nums.length <= 3 * 10<sup>4</sup></code></li>
<li><code>nums[i]</code> 不是 <code>0</code> 就是 <code>1</code></li>
<li><code>0 <= goal <= nums.length</code></li>
</ul>
<div><div>Related Topics</div><div><li>数组</li><li>哈希表</li><li>前缀和</li><li>滑动窗口</li></div></div>\n<div><li>👍 157</li><li>👎 0</li></div>

File diff suppressed because one or more lines are too long

File diff suppressed because one or more lines are too long