给定4个组合A,B,C,D,这四个组合里面的元素数目相等。要求找出所有的序列(a, b, c, d),四个数分别来自A,B,C,D四个组合,使得a + b + c + d = 0,输出这样的序列的总数。若两个满足条件的序列是完全相同的,它们也算是不同的序列,因为A,B,C,D中可能含有相同的元素。
解题思路很简单,取所有的(a, b)序列进行哈希,以a+b作为键,分别映射到非负数和负数两个哈希表上,若有重复的(a, b),则只需要记录次数,不需要再开辟结点存储。
然后对所有的(c, d)序列,以c+d作为键进行查找,若c+d为正则查找对应的负数哈希表,若为负则查找对应的非负数哈希表,把符合a+b+c+d=0的所有结点的所有次数相加就得到最后的答案了。
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
const int N = 9000005;
const int M = 4005;
const int H = 4000007;
int n;
long long ans;
int A[M], B[M], C[M], D[M];
struct Node
{
int a;
int b;
int cnt;
int next;
};
Node nodeP[N];
Node nodeN[N];
int curP, curN;
int hashTableP[H];
int hashTableN[H];
void