本题解题步骤:
- 首先定义 一个unordered_map,key放a和b两数之和,value 放a和b两数之和出现的次数。
- 遍历大A和大B数组,统计两个数组元素之和,和出现的次数,放到map中。
- 定义int变量count,用来统计 a+b+c+d = 0 出现的次数。
- 在遍历大C和大D数组,找到如果 0-(c+d) 在map中出现过的话,就用count把map中key对应的value也就是出现次数统计出来。
- 最后返回统计值 count 就可以了
struct hashTable {
int key;
int val;
UT_hash_handle hh;
};
int fourSumCount(int* A, int ASize, int* B, int BSize, int* C, int CSize, int* D, int DSize) {
struct hashTable* hashtable = NULL;
for (int i = 0; i < ASize; ++i) {
for (int j = 0; j < BSize; ++j) {
int ikey = A[i] + B[j];
struct hashTable* tmp;
HASH_FIND_INT(hashtable, &ikey, tmp);
if (tmp == NULL) {
struct hashTable* tmp = malloc(sizeof(struct hashTable));
tmp->key = ikey, tmp->val = 1;
HASH_ADD_INT(hashtable, key, tmp);
} else {
tmp->val++;
}
}
}
int ans = 0;
for (int i = 0; i < CSize; ++i) {
for (int j = 0; j < DSize; ++j) {
int ikey = -C[i] - D[j];
struct hashTable* tmp;
HASH_FIND_INT(hashtable, &ikey, tmp);
if (tmp != NULL) {
ans += tmp->val;
}
}
}
return ans;
}