leetcode 454. 四数相加 II

题目链接
思路:分组+哈希表
分析:四个树状A、B、C、D,将AB分为一组,CD分为一组。
首先计算AB一共可以加出多少个不同的数字来,并且记录每个数字有多少种情况可以加出来。(记录在哈希表种)
然后再计算0减去CD的值相加,并且去哈希表中找,看有几种情况。
也就是将A+B记在哈希表中,然后在哈希表中寻找-(C+D),看有多少种情况,然后进行累加即可得到答案。
代码:

class Solution {
    public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
        int len = nums1.length;
        //key是记录A+B,val是记录多少用不同的情况
        HashMap<Integer, Integer> map = new HashMap<>();
        int res=0;
        for(int i=0;i<len;i++){
            for(int j=0;j<len;j++){
            	//将情况+1
                map.put(nums1[i]+nums2[j],map.getOrDefault(nums1[i]+nums2[j],0)+1);
            }
        }
        for(int i=0;i<len;i++){
            for(int j=0;j<len;j++){
            	//-(C+D)在哈希表中寻找。并且累加答案
                res = res + map.getOrDefault(-nums3[i]-nums4[j],0);
            }
        }
        return res;
    }
}

你可能感兴趣的:(算法,leetcode,算法)