给你四个整数数组 nums1、nums2、nums3 和 nums4 ,数组长度都是 n ,请你计算有多少个元组 (i, j, k, l) 能满足:
- 0 <= i, j, k, l < n
- nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0
思路:如果要暴力,那么时间复杂度将会是 O ( n 4 ) O(n^4) O(n4);可以通过两两数组分别遍历,那么就能将时间复杂度降到 O ( n 2 ) O(n^2) O(n2)。具体的思路是:
- 先遍历nums1, nums2,将其所有可能的和都添加到HashMap中,同时记录重复出现的次数;
- 再遍历nums3, nums4,查找HashMap中是否存在(0-i-j)这个Key,该Key对应的Value就是能与(i,j)组成符合要求的四元组的情况数目。
解答:
class Solution {public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {HashMap<Integer,Integer> hm = new HashMap<>();int count = 0;// 将num1, num2数组两数之和及出现频率存储到HashMapfor(int i: nums1) for(int j: nums2)if(hm.containsKey(i+j))hm.put(i+j, hm.get(i+j)+1);elsehm.put(i+j, 1);// 遍历计算num3,num4所有的和for(int i: nums3)for(int j: nums4)if(hm.containsKey(0-i-j))count += hm.get(0-i-j);return count;}
}