454、四数之和（哈希表）

 

题目：

给定四个包含整数的数组列表 A , B , C , D ,计算有多少个元组 (i, j, k, l) ，使得 A[i] + B[j] + C[k] + D[l] = 0。

为了使问题简单化，所有的 A, B, C, D 具有相同的长度 N，且 0 ≤ N ≤ 500 。所有整数的范围在 -2^28 到 2^28 - 1 之间，最终结果不会超过 2^31 - 1。

方法：哈希表法

思路：将四个数组两两分组，记录其中两个数组之和以及他们 和的出现次数，并将他们以和的值为key，出现次数为value存放在哈希表中，

另外两个数组和的值作为key去寻找哈希表中是否存在key使得四个数组的和为0，即map.containKey（0-nums3[i]+nums4[j]）

如果找到了符合条件的key值，就更新出现次数，遍历完后返回最后的出现次数

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

public class Sishuzhihe2 {
    //首先定义一个hashmap，key存放a和b两数之和，value放a和b两数之和出现的次数
    //遍历A，B两数组，统计量数组元素之和出现的次数，并放入map
    //定义int变量count，统计a+b+c+d=0出现的次数
    //在遍历C和D数组，如果找到0-（c+d）在map中出现过的话，就用count把map中key对应的value也就是出现次数统计出来
    //返回count
    public int fourSumCount(int [] nums1,int[] nums2,int[] nums3,int[] nums4){
        Map<Integer,Integer> map = new HashMap<>();
        int temp;
        int res = 0;
        //统计两个数组中的元素之和，同时统计出现的次数，放入map
        for (int i:nums1) {
            for (int j:nums2) {
                temp = i+j;
                if(map.containsKey(temp)){
                    map.put(temp,map.get(temp)+1);
                }else {
                    map.put(temp,1);
                }
            }
        }

        //统计剩下两个数组的元素和，在map中找到是否存在key使得,相加为0，同时记录出现次数
        for (int i : nums3) {
            for (int j : nums4) {
                temp= i+j;
                if (map.containsKey(0-temp)){
                    res+= map.get(0-temp);
                }
            }
        }
        return res;
    }

    public static void main(String[] args) {
        int [] a = {1,2};
        int [] b = {-2,-1};
        int [] c = {-1,2};
        int [] d = {0,2};
        Sishuzhihe2 sishuzhihe2 = new Sishuzhihe2();
        int res = sishuzhihe2.fourSumCount(a,b,c,d);
        System.out.println(res);
    }
}