454. 四数相加 II

做题思路 or 感想:

  1,用一个哈希表储存  A  +  B 的各种和的个数,把题目转化为两个数(A + B 和 C + D)之和,知道这个了以后思路就开阔很多了

  2,因为这里是四个数组,所以不需要查重,直接搜,find 搜到直接符合条件就好了

 1 class Solution {
 2 public:
 3     int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
 4         unordered_map<int, int>HashMap;     //用来记录 A + B数组的各种和的个数
 5         int result = 0;
 6         for (int i = 0; i < nums1.size(); i++) {
 7             for (int j = 0; j < nums2.size(); j++) {    //记录 A + B 的和
 8                 int sum = nums1[i] + nums2[j];
 9                 HashMap[sum]++;
10             }
11         }
12         for (int i = 0; i < nums3.size(); i++) {
13             for (int j = 0; j < nums4.size(); j++) {    //因为这里是四个不同的数组,所以不需要查重,如果都是一个数组,则要查重了!
14                 //由于 A + B = - C - D,故这里直接找就好了
15                 if (HashMap.find(-nums3[i] - nums4[j]) != HashMap.end()) {  
16                     auto it = HashMap.find(-nums3[i] - nums4[j]);
17                     result += (*it).second;
18                 }
19             }
20         }
21         return result;
22     }
23 };

 

posted @ 2022-03-23 19:55  北原春希  阅读(46)  评论(0)    收藏  举报