leetcode 454. 四数相加 II

454. 四数相加 II - 力扣(LeetCode) (leetcode-cn.com)icon-default.png?t=M0H8https://leetcode-cn.com/problems/4sum-ii/先遍历前两个数组元素之间的和,然后记录到Map中(key是和,value是和出现的次数)。

然后遍历后两个数组元素之间的和,判断这个和的相反数书否在Map中,若在,count 就加上此相反数出现的次数。

class Solution {
    public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
         Map map = new HashMap<>();
        //先计算前两个数组的和,并统计每个和出现的次数
        for(int i:nums1){
            for(int j:nums2){
                if (map.containsKey(i+j))
                    map.put(i+j,map.get(i+j)+1);
                else
                    map.put(i+j,1);
            }
        }
        //计算后两个数组的和,count+=和的相反数出现的次数
        int count = 0;
        for(int i:nums3){
            for(int j:nums4){
                if(map.containsKey(-(i + j))){
                    count+=map.get(-(i+j));
                }
            }
        }

        return count;
    }
}

你可能感兴趣的:(LeetCode,leetcode,算法,职场和发展,java)