【LeetCode算法修炼+动画演示】【数组的和】——X.未排序数组中累加和为给定值的最长子数组系列问题、525. 连续数组

X.未排序数组中累加和为给定值的最长子数组系列问题

题目来源:《程序员代码面试指南：IT 名企算法与数据结构题目最优解》

给定一个无序数组arr，其中元素可正、可负、可0，给定一个整数k。求arr所有的子数组中累加和为k的最长子数组长度。

补充题目

1. 给定一个无序数组arr，其中元素可正、可负、可0。求arr所有的子数组中正数与负数个数相等的最长子数组长度
2. 给定一个无序数组arr，其中元素只是1或0。求arr所有的子数组中0和1个数相等的最长子数组长度

效率

时间复杂度:$O(n)$
空间复杂度:$O(n)$

要点

第一次解说，如有问题请指正、请谅解:

带解说

纯动画演示：

纯动画

HashMap解法

在遍历过程中使用一个 HashMap记录当前数组和的状态

代码片段

class Solution {
    public static void main(String[] args) {
        Solution solution = new Solution();
        int[] arr = {2, -3, 4, 3, 3};
        int maxLength = solution.maxLength(arr, 6);
        System.out.println(maxLength);
    }
    public int maxLength(int[] arr, int k) {
        if (arr == null || arr.length == 0) {
            return 0;
        }
        HashMap<Integer, Integer> hashMap = new HashMap<>();//创建一个储存当前 sum 以及对应的 pos(这里是i)。
        int len = 0;
        int sum = 0;
        hashMap.put(0, -1);//为hashMap填充一个起始点。
        for (int i = 0; i < arr.length; i++) {
            sum += arr[i];
            if (hashMap.containsKey(sum - k)) {//如果能在之前找到一个合适的值，取当前最大长度。
                Integer integer = hashMap.get(sum - k);
                len = Math.max(len, i - integer);
            }
            if (!hashMap.containsKey(sum)) {//如果当前的sum还没有记录，将它以及pos记录到HashMap
                hashMap.put(sum, i);
            }
        }
        return len;
    }
}

补充题目

1. 给定一个无序数组arr，其中元素可正、可负、可0。求arr所有的子数组中正数与负数个数相等的最长子数组长度

   // 可以把 正的都当做1，负的当做 -1 ，将k当做0
   public int maxLength(int[] arr) {
   	int k = 0;//差异代码 1
       for (int i = 0; i < arr.length; i++) {
       	//差异代码 2
           if (arr[i] > 0) {
               sum += 1;
           } else if (arr[i] < 0) {
               sum += -1;
           }
       }
   }
   

2. 给定一个无序数组arr，其中元素只是1或0。求arr所有的子数组中0和1个数相等的最长子数组长度

   // 可以把 正的都当做1，0都的当做 -1 ，将k当做0
   public int maxLength(int[] arr) {
   	int k = 0;//差异代码 1
       for (int i = 0; i < arr.length; i++) {
       	//差异代码 2
           if (arr[i] == 0) {
              	sum += -1;
          	} else {
              	sum += 1;
          	}
       }
   }
   

---

525. 连续数组

原题链接
给定一个二进制数组, 找到含有相同数量的 0 和 1 的最长连续子数组（的长度）。
示例 1:

输入: [0,1]
输出: 2
说明: [0, 1] 是具有相同数量0和1的最长连续子数组

示例 2:

输入: [0,1,0]
输出: 2
说明: [0, 1] (或 [1, 0]) 是具有相同数量0和1的最长连续子数组。

要点

这一题在 leetcode 上的难度是hard，但是使用我们上文提到的 HashMap 能快速解决问题。

HashMap解法

代码片段

class Solution {
    public int findMaxLength(int[] nums) {
        if (nums==null||nums.length==0){
            return 0;
        }
        HashMap<Integer, Integer> hashMap = new HashMap<>();
        hashMap.put(0,-1);
        int len = 0;
        int sum = 0;
        for(int i=0;i<nums.length;i++){
            if (nums[i]==0){
                sum-=1;
            }else{
                sum += 1;
            }
            if (hashMap.containsKey(sum)){
                len = Math.max(len, i - hashMap.get(sum));
            }else{
                hashMap.put(sum,i);
            }
        }
        return len;
    }
}

代码解释

我再这里将 k=0 这个初始值去掉，因为 sum-0=sum。
合并了 if else判断key是否存在，因为此时 sum-k = sum。
去掉了 hashMap.get赋值的操作。
这是为了节省空间在 leetcode能更好的通过。

---

😁😁😁制作动画过程不易，请大家顺手点赞收藏咯 谢谢~😁😁😁 有其它题目不理解的也可以一起学习，如有错误欢迎指出~