454. 四数相加 II

给你四个整数数组 nums1、nums2、nums3 和 nums4 ，数组长度都是 n ，请你计算有多少个元组 (i, j, k, l) 能满足：

• 0 <= i, j, k, l < n
• nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0

• class Solution {
  public:
      int fourSumCount(vector<int>& A, vector<int>& B, vector<int>& C, vector<int>& D) {
          unordered_map<int, int> umap; //key:a+b的数值，value:a+b数值出现的次数
          // 遍历大A和大B数组，统计两个数组元素之和，和出现的次数，放到map中
          for (int a : A) {
              for (int b : B) {
                  umap[a + b]++;
              }
          }
          int count = 0; // 统计a+b+c+d = 0 出现的次数
          // 在遍历大C和大D数组，找到如果 0-(c+d) 在map中出现过的话，就把map中key对应的value也就是出现次数统计出来。
          for (int c : C) {
              for (int d : D) {
                  if (umap.find(0 - (c + d)) != umap.end()) {
                      count += umap[0 - (c + d)];
                  }
              }
          }
          return count;
      }
      int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
          multimap<int, vector<pair<int, int>>> mp1;
          multimap<int, vector<pair<int, int>>> mp2;
          int res = 0;
          for (int i = 0; i < nums1.size(); i++)
          {
              for (int j = 0; j < nums1.size(); j++)
              {
                  int sum1 = nums1[i] + nums2[j];
                  int sum2 = nums3[i] + nums4[j];
                  vector<pair<int, int>> vec_sum1;
                  vector<pair<int, int>> vec_sum2;
                  vec_sum1.emplace_back(nums1[i], nums2[j]);
                  vec_sum2.emplace_back(nums3[i], nums4[j]);
                  mp1.insert(multimap<int, vector<pair<int, int>>>::value_type(sum1, vec_sum1));
                  mp2.insert(multimap<int, vector<pair<int, int>>>::value_type(sum2, vec_sum2));
              }
          }
          for (const auto& p : mp1)
          {
              auto nums = mp2.count(0 - p.first);
              auto iter = mp2.find(0 - p.first);
              if (iter != mp2.end())
              {
                  res = res + nums;
              }
          }
          return res;
      }
  };