网站风格南宁个人网站建设

题号454

给你四个整数数组 nums1、nums2、nums3 和 nums4 ，数组长度都是 n ，请你计算有多少个元组 (i, j, k, l) 能满足：

0 <= i, j, k, l < n
nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0

解法

class Solution {public int fourSumCount(int[] A, int[] B, int[] C, int[] D) {Map<Integer,Integer>map=new HashMap<>();int n=A.length;for(int i=0;i<n;i++){for(int j=0;j<n;j++){map.put(A[i]+B[j],map.getOrDefault(A[i]+B[j],0)+1);}}int count=0;for(int k=0;k<n;k++){for(int l=0;l<n;l++){if(map.containsKey(-(C[k]+D[l])))count+=map.get(-(C[k]+D[l]));}}return count;}
}

思路

将AB CD分为两组，先遍历AB 将两者之和作为Key，出现的次数作为value存进HashMap中，然后再遍历CD，若出现-Key，则说明可以与AB中的元素相加为0，此时count+=value，即可记录所有相加为0的次数了