day7 leetcode 454. 四数相加 II

最新推荐文章于 2024-07-21 15:36:00 发布

gxinyang

最新推荐文章于 2024-07-21 15:36:00 发布

阅读量41

点赞数

文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/Agxyzx/article/details/131544504

版权

思路

相当于A+B+C+D=0,求ABCD多少种可能性，每个数组数据量10^2,刚开始的时候我是遍历A,B,C，然后看-A-B-C在不在D对应的Map中，然后将他们元素对应的次数相乘即为答案，时间超限，所以，考虑只遍历AB和CD，然后看他们的和是否为相反数。

题目描述

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

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

代码

package Hash;

import java.util.HashMap;
import java.util.Map;
import java.util.Scanner;

public class _454四数相加 {
	 
		  public static int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
				
		HashMap<Integer,Integer> map1=new HashMap<Integer,Integer>();
		for(int i=0;i<nums1.length;i++) {
			map1.put(nums1[i], map1.getOrDefault(nums1[i], 0)+1);
		}
		HashMap<Integer,Integer> map2=new HashMap<Integer,Integer>();
		for(int i=0;i<nums2.length;i++) {
			map2.put(nums2[i], map2.getOrDefault(nums2[i], 0)+1);
		}
		
		HashMap<Integer,Integer> summap12=new HashMap<Integer,Integer>();
		for(Map.Entry<Integer, Integer> map11: map1.entrySet()) {
			for(Map.Entry<Integer, Integer> map22:map2.entrySet()) {
				summap12.put(map11.getKey()+map22.getKey(), summap12.getOrDefault(map11.getKey()+map22.getKey(), 0)+map11.getValue()*map22.getValue());
			}
		}
		
		
		HashMap<Integer,Integer> map3=new HashMap<Integer,Integer>();
		for(int i=0;i<nums3.length;i++) {
			map3.put(nums3[i], map3.getOrDefault(nums3[i], 0)+1);
		}
		HashMap<Integer,Integer> map4=new HashMap<Integer,Integer>();
		for(int i=0;i<nums4.length;i++) {
			map4.put(nums4[i], map4.getOrDefault(nums4[i], 0)+1);
		}

		HashMap<Integer,Integer> summap34=new HashMap<Integer,Integer>();
		for(Map.Entry<Integer, Integer> map33: map3.entrySet()) {
			for(Map.Entry<Integer, Integer> map44:map4.entrySet()) {
				summap34.put(map33.getKey()+map44.getKey(), summap34.getOrDefault(map33.getKey()+map44.getKey(), 0)+map33.getValue()*map44.getValue());
			}
		}
		
		/*for(Map.Entry<Integer, Integer> map12:summap12.entrySet()) {
			System.out.println(map12.getKey()+" "+map12.getValue());
		}
		System.out.println("********");
		for(Map.Entry<Integer, Integer> map34:summap34.entrySet()) {
			System.out.println(map34.getKey()+" "+map34.getValue());
		}*/
		int count=0;
		for(Map.Entry<Integer, Integer> map12:summap12.entrySet()) {
			if(summap34.containsKey(-map12.getKey())) {
				count=count+map12.getValue()*summap34.get(-map12.getKey());
			}
		}
		
		//System.out.println(count);
		  return count;

	    }
	public static void main(String[] args) {
		Scanner sc=new Scanner(System.in);
		while(sc.hasNext()) {
			int n1=sc.nextInt();
			int nums1[]=new int[n1];
			for(int i=0;i<n1;i++) {
				nums1[i]=sc.nextInt();
			}
			int n2=sc.nextInt();
			int nums2[]=new int[n2];
			for(int i=0;i<n2;i++) {
				nums2[i]=sc.nextInt();
			}
			int n3=sc.nextInt();
			int nums3[]=new int[n3];
			for(int i=0;i<n3;i++) {
				nums3[i]=sc.nextInt();
			}
			int n4=sc.nextInt();
			int nums4[]=new int[n4];
			for(int i=0;i<n4;i++) {
				nums4[i]=sc.nextInt();
			}
			fourSumCount(nums1,nums2,nums3,nums4);
			
		}

	}

}