leetcode_middle_18_454. 4Sum II

来源:互联网 发布:免费视频编辑制作软件 编辑:程序博客网 时间:2024/06/07 01:05
题意:

给四个长度相等的整数数组,每个数组取一个数相加,等于0.这样的组合有多少个?


分析:

我们要全部搜索遍历的话需要四重循环,有没有什么办法不四重循环呢,首先想到的是重复出现的可以用map来存放它出现的次数,但是这个并不能减少循环次数。继续思考,没什么头绪,我们来观察题目特征。我们要的是和为a+b+c+d等于0意味着四重循环 ,可以推出:a+b+c = -d , a+b =  -c - d...是不是意味着三重循环加一重循环,二重循环加二重循环呢? 我们继续整理思路,选取最好的两个二重循环:

public class Solution {    public int fourSumCount(int[] A, int[] B, int[] C, int[] D) {        Map<Integer, Integer> map = new HashMap<>();        int count = 0;        for(int i=0; i<A.length; i++) {            for(int j=0; j<B.length; j++) {                 int sum1 = A[i] + B[j];                 if(map.containsKey(sum1)){                    map.put(sum1, map.get(sum1)+1);                }else{                    map.put(sum1, 1);                }            }        }                for(int i=0; i<C.length; i++) {            for(int j=0; j<D.length; j++) {                 int sum2 = -C[i] - D[j];                 if(map.containsKey(sum2)){                    count += map.get(sum2);                }            }        }        return count;    }}


0 0
原创粉丝点击