【LeetCode 14】454.四数相加 II-阿里云开发者社区

【LeetCode 14】454.四数相加 II

2024-10-14 13

版权

本文内容由阿里云实名注册用户自发贡献，版权归原作者所有，阿里云开发者社区不拥有其著作权，亦不承担相应法律责任。具体规则请查看《阿里云开发者社区用户服务协议》和《阿里云开发者社区知识产权保护指引》。如果您发现本社区中有涉嫌抄袭的内容，填写侵权投诉表单进行举报，一经查实，本社区将立刻删除涉嫌侵权内容。

简介： 【LeetCode 14】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;
    }
};