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 };