其他分享
首页 > 其他分享> > 四数相加

四数相加

作者:互联网

https://leetcode-cn.com/problems/4sum-ii/
该题大意是说给我们四个数组,判断从每个数组中各取一个数相加能否等于零,因为是四数相加,所以采用暴力解法的复杂度会非常大
因此,我们采用哈希表法
因为数组不需要有序,为了读入数据更加快捷,我们采用unordered_map来存储哈希表
首先,遍历前两个数组,获得每两个元素相加的和存储到哈希表中
然后,遍历后两个数组,前两个数组与后两个数组的和即a+b+c+d,即前两个数组的和在等于-c-d 时四数之和为零,当满足条件时,将结果累计,因为umap[]数组存储的是相加等于某一相同值的个数,所以累计的是umap[-c-d]
下面是代码

unordered_map<int,int>umap;
        for(int i:nums1)
            for(int j:nums2)
                umap[i+j]++;
        int count=0;
        for(int i:nums3)
            for(int j:nums4)
                if(umap.find(-(i+j))!=umap.end())count+=umap[-(i+j)];
        return count;

标签:四数,数组,int,相加,umap,哈希
来源: https://blog.csdn.net/m0_51054450/article/details/115867222