题意:给定4个n元素集合,要求从每个集合中选择一个数,使得A+B+c+d==0,问存在多少种方法。
思路:枚举+hash判断。直接枚举4次方的算法会超,那么只需要枚举ab,然后在c+d的和中查找等于-(a+b)的值(用mp或hash,n2logn的算法)。
Code:
#include <bits/stdc++.h>
using namespace std;const int N=4005;
int a[N],b[N],c[N],d[N];
struct Hash_map
{static const int m=0x7fffff;int p[m+1],q[m+1];void cls(){for (int i=0;i<=m;i++) q[i]=0;}int& operator[] (int k){int i;for (i=k&m;q[i]&&p[i]!=k;i=(i+1)%m);p[i]=k;return q[i];}
}has;
int main()
{int T,n;scanf("%d",&T);while (T--){scanf("%d",&n);for (int i=1;i<=n;i++) scanf("%d %d %d %d",&a[i],&b[i],&c[i],&d[i]);int ans=0;has.cls();for (int i=1;i<=n;i++)for (int j=1;j<=n;j++)has[a[i]+b[j]]++;for (int i=1;i<=n;i++)for (int j=1;j<=n;j++)ans+=has[-(c[i]+d[j])];printf("%d\n",ans);if (T) puts("");}
}