脚本宝典收集整理的这篇文章主要介绍了四数相加II Java,脚本宝典觉得挺不错的,现在分享给大家,也给大家做个参考。
1.使用HashMap
class Solution {
public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
//统计前两个数组中的元素之和(key),以及出现次数(valuevalue)
Map<Integer,Integer> map = new HashMap<>();
int res = 0;
for(int i : nums1){
for(int j : nums2){
int sum1 = i + j;
if(map.containsKey(sum1)){
map.put(sum1,map.get(sum1)+1);
}else{
map.put(sum1,1);
}
}
}
//统计剩余的两个元素的和,在map中找是否存在相加为0的情况,同时记录次数
for(int m : nums3){
for(int n : nums4){
int sum2 = m + n;
if(map.containsKey(0-sum2)){
res += map.get(0-sum2);
}
}
}
return res;
}
}
优化
class Solution {
public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
int res=0;
HashMap<Integer,Integer> map=new HashMap<>();
for(int a:nums1){
for(int b:nums2){
map.put(a+b,map.getOrDefault(a+b,0)+1);
}
}
// 先统计了前两个的和出现次数
for(int c:nums3){
for(int d:nums4){
int temp=map.getOrDefault(-(c+d),-1);
if(temp!=-1)
res+=temp;
}
}
//再在后两个数组中元素的和的相反数中,在map里寻找,
//是否包含其相反数,包含的话就说明有和为0的四个数,计数器计数
//否则,循环进行下一次,寻找下一个和相反数
//也可以先统计前三个数组各个元素的和的出现次数。
return res;
}
}
以上是脚本宝典为你收集整理的四数相加II Java全部内容,希望文章能够帮你解决四数相加II Java所遇到的问题。
本图文内容来源于网友网络收集整理提供,作为学习参考使用,版权属于原作者。
如您有任何意见或建议可联系处理。小编QQ:384754419,请注明来意。