【LeetCode 14】454.四数相加 II

最新推荐文章于 2022-07-07 22:59:55 发布

CodeLinghu

最新推荐文章于 2022-07-07 22:59:55 发布

阅读量498

点赞数

分类专栏： LeetCode 文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/weixin_43891901/article/details/122706690

版权

LeetCode 专栏收录该内容

71 篇文章 2 订阅

订阅专栏

454.四数相加 II

文章目录

454.四数相加 II
一、题意
二、思考过程
三、完整代码

一、题意

二、思考过程

2.1定义一个unordered_map

/*
        key:a+b的数值
        value：a+b数值出现的次数
       */
        unordered_map<int,int> umap;

2.2遍历nums1、nums2数组，统计两个数组的元素之和及出现的次数，并放到map中

for(int a:nums1){xxx};这里的for循环是增强型for循环。意思是把nums1数组中的每一个元素取出来存到变量a中

   //遍历nums1和nums2数组，统计两个数组元素之和以及出现次数，并放到map中
        for(int a:nums1){
            for(int b:nums2){
                umap[a+b]++;
            }
        }

2.3定义int类型的变量count,用来统计a+b+c+d=0出现的次数

int count=0;//统计a+b+c+d=0出现的次数

2.4在遍历nums3、nums4数组，如果(0-(c+d)）在map中出现，就用count来统计两数之和出现次数

    //遍历nums3和nums4数组，如果0-(c+d)在map中出现过
        //统计map中key对应的value值，也就是两数之和出现的次数
        for(int c:nums3){
            for(int d:nums4){
                if(umap.find(0-(c+d))!=umap.end()){
                    count+=umap[0-(c+d)];
                }
            }
        }

2.5返回统计值count

return count;

三、完整代码

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
       /*
        key:a+b的数值
        value：a+b数值出现的次数
       */
        unordered_map<int,int> umap;

        //遍历nums1和nums2数组，统计两个数组元素之和以及出现次数，并放到map中
        for(int a:nums1){
            for(int b:nums2){
                umap[a+b]++;
            }
        }
        int count=0;//统计a+b+c+d=0出现的次数

        //遍历nums3和nums4数组，如果0-(c+d)在map中出现过
        //统计map中key对应的value值，也就是两数之和出现的次数
        for(int c:nums3){
            for(int d:nums4){
                if(umap.find(0-(c+d))!=umap.end()){
                    count+=umap[0-(c+d)];
                }
            }
        }
        return count;

    }
};