Subarray 系列汇总

最新推荐文章于 2022-10-20 08:00:00 发布

Pengyu Che

最新推荐文章于 2022-10-20 08:00:00 发布

阅读量385

点赞数

分类专栏： Leetcode 文章标签： leetcode

本文链接：https://blog.csdn.net/Nighticerain/article/details/116517819

版权

这篇博客汇总了关于求解子数组和子矩阵之和的问题，包括LintCode 138的子数组之和，LeetCode 560的和为K的子数组，以及LintCode 405的和为零的子矩阵。文中提供了样例输入和输出，以及解题思路，特别提到了使用前缀和的方法。同时，对解题的复杂度提出了挑战，要求在O(n^3)的时间内完成。

摘要由CSDN通过智能技术生成

LintCode 138 子数组之和

描述
给定一个整数数组，找到和为 0 的子数组。你的代码应该返回满足要求的子数组的起始位置和结束位置

注：至少有一个子数组的和为 0

样例 1:

输入: [-3, 1, 2, -3, 4]
输出: [0,2] 或 [1,3]
样例解释：返回任意一段和为0的区间即可。

样例 2:

输入: [-3, 1, -4, 2, -3, 4]
输出: [1,5]

解题思路：前缀和

参考代码：

public class Solution {
   
    /**
     * @param nums: A list of integers
     * @return: A list of integers includes the index of the first number and the index of the last number
     */
    public List<Integer> subarraySum(int[] nums) {
   
        int n = nums.length;
        int[] sum = new int[n+1];

        Map<Integer, Integer> map = new HashMap<>();
        map.put(0,0);
        List<Integer> ans = new ArrayList<>();
        for (int i=1; i<=n; i++) {
   
            sum[i] = sum[i-1]+ nums[i-1];
            if (map.containsKey(sum[i])) {
   
                ans.add(map.get<