poj 2785

一道经典的问题，就是找四个数的和为零.

解法：

先把前两列的和算出来。再把后两列两两和的相反数算出来。利用hash表，或者排序+二分解决。

1.排序+二分法

#include <iostream>
#include <algorithm>
using namespace std;

int n,a[4040],b[4040],c[4040],d[4040],ans,ab[4040*4040],cd[4040*4040];
int main()
{
	int i,j,k,pab=0,pcd=0;
	scanf("%d",&n);
	for(i=0;i<n;i++)
		scanf("%d%d%d%d",&a[i],&b[i],&c[i],&d[i]);
	for(i=0;i<n;i++)
		for(j=0;j<n;j++)
		{
			ab[pab++]=a[i]+b[j];
			cd[pcd++]=-(c[i]+d[j]);
		}
	sort(cd,cd+pcd);
	for(i=0;i<pab;i++)
	{
		int l=0,r=pcd-1,mid;
		while(l<=r)
		{
			mid=(l+r)/2;
			if(ab[i]==cd[mid])
			{
				ans++;
				for(k=mid+1;k<pcd;k++)
					if(ab[i]==cd[k])
						ans++;
					else
						break;
				for(k=mid-1;k>=0;k--)
					if(ab[i]==cd[k])
						ans++;
					else
						break;
				break;
			}
			else if(ab[i]<cd[mid])
				r=mid-1;
			else
				l=mid+1;
		}
	}
	printf("%d/n",ans);
}

2.hash表法

• #include <iostream>
  #define MAX 1000000000
  #define size 20345677
  #define key 745
  using namespace std;
  
  int n,a[4040],b[4040],c[4040],d[4040],ans;
  int hash[size],sum[size];
  void Insert(int num)
  {
  	int tmp=num;
  	num=(num+MAX)%size;
  	while(hash[num]!=MAX && hash[num]!=tmp)
  		num=(num+key)%size;
  	hash[num]=tmp;
  	sum[num]++;
  }		
  int Find(int num)
  {
  	int tmp=num;
  	num=(num+MAX)%size;
  	while(hash[num]!=MAX && hash[num]!=tmp)
  		num=(num+key)%size;
  	if(hash[num]==MAX)
  		return 0;
  	else
  		return sum[num];
  }	
  int main()
  {
  	int i,j;
  	scanf("%d",&n);
  	for(i=0;i<n;i++)
  		scanf("%d%d%d%d",&a[i],&b[i],&c[i],&d[i]);
  	for(i=0;i<size;i++)
  		hash[i]=MAX;
  	for(i=0;i<n;i++)
  		for(j=0;j<n;j++)
  			Insert(a[i]+b[j]);
  	for(i=0;i<n;i++)
  		for(j=0;j<n;j++)
  			ans+=Find(-(c[i]+d[j]));
  	printf("%d/n",ans);
  }