题解 | #子群的标签和#

子群的标签和

https://www.nowcoder.com/practice/4058a95b317f4f0e87e5a1bfa6db9aad

知识点:前缀和,哈希表

对于判断子数组的和,可以通过前缀和和哈希表的方式,将每一个前缀和作为key,下标作为value,每次获取到新的前缀和,可以在map中判断是否存在preSum-k的前缀和,也就是二者的区间和为k,若存在,则保存下来该区间的子数组,加入到TreeSet中。此处使用的是TreeSet,目的是按字典序排列,且去除相同子数组。在TreeSet中指定比较器,二者首元素不同时,比较大小,首元素相同时,比较二者的长度,以筛选出不同的子数组。

Java题解如下

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param nums int整型一维数组 
     * @param k int整型 
     * @return int整型二维数组
     */
    public int[][] subarraySum (int[] nums, int k) {
        // write code here
        int n = nums.length;
        Map<Integer, Integer> map = new HashMap<>();
        int preSum = 0;
        map.put(0, -1);
        TreeSet<int[]> set = new TreeSet<>((o1, o2) -> 
            {
                if(o1[0] != o2[0]) {
                    return o1[0] - o2[0];
                }else {
                    return o1.length - o2.length;   
                }
            });
        for(int i = 0; i < n; i++) {
            preSum += nums[i];
            if(map.containsKey(preSum - k)) {
                int start = map.get(preSum - k) + 1;
                int[] arr = new int[i - start + 1];
                for(int j = start; j <= i; j++) {
                    arr[j - start] = nums[j];
                }
                set.add(arr);
            }
            map.put(preSum, i);
        }
        int[][] res = new int[set.size()][];
        set.toArray(res);
        return res;
    }
}

全部评论

相关推荐

评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务