一、题意
二、思考过程
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; } };
