poj 2785 4 Values…

来源:互联网 发布:大津算法 matlab 编辑:程序博客网 时间:2024/06/03 20:29
题意:
     给你4个集合a,b,c,d
     从其中各选一个值,使其和等于0
     一共有多少种选法。每个集合最多有4000个元素。
这个题目和之前发的1840基本是一个思路
如果裸枚举的话,是n^4。然后就考虑分成两边枚举,先枚举两个值的和,存在哈希表中,然后在枚举另外两个值的和,查哈希表中它的相反数。

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <math.h>
using namespace std;
const int maxn=10000000;
long long ans;
int hash[maxn],lon;
struct
{
    int tmp,next,sum;
}data[maxn*2];

int a[4444][5];

int insert(int a,int b)
{
    int txt=fabs(a+b);
    txt%=maxn;
    for(intk=hash[txt];k!=-1;k=data[k].next)
    {
       if(a+b==data[k].tmp)
       {
          data[k].sum++;
          return(0);
       }
    }
   data[++lon].tmp=a+b;
    data[lon].sum=1;
   data[lon].next=hash[txt];
    hash[txt]=lon;
}

int check(int a,int b)
{
    int txt=fabs(a+b);
    txt%=maxn;
    for(intk=hash[txt];k!=-1;k=data[k].next)
    {
       if(-data[k].tmp==a+b)
       {
          ans+=data[k].sum;
          return(0);
       }
    }
}



int main()
{
    int n;
   while(scanf("%d",&n)!=EOF)
    {
       lon=0;
       ans=0;
       memset(hash,-1,sizeof(hash));
       for(int i=1;i<=n;i++)
       for(int j=1;j<=4;j++)
       scanf("%d",&a[i][j]);
       for(int i=1;i<=n;i++)
       for(int j=1;j<=n;j++)
       {
          insert(a[i][1],a[j][2]);
       }
       for(int i=1;i<=n;i++)
       for(int j=1;j<=n;j++)
       {
          check(a[i][3],a[j][4]);
       }
       printf("%lld\n",ans);
    }
    return 0;
}


原创粉丝点击