【leetcode】454. 4Sum II

最新推荐文章于 2021-12-25 19:07:23 发布

AXIMI

最新推荐文章于 2021-12-25 19:07:23 发布

阅读量156

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/AXIMI/article/details/104695046

版权

leetcode 专栏收录该内容

257 篇文章 2 订阅

订阅专栏

解题思路

四个数组中，找出组合之和为0的个数。一般想到的是4个for循环，时间复杂度O(n^4)，耗时太久了。可以考虑牺牲空间换取时间。将A B两个数组中数字的组合值用map保存起来，然后再遍历C D数组。综合时间复杂度O(N^2).

提交代码

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