POJ 2785 4 Values whose Sum is 0

题目传送门

【题意】题意很简单,给了n*4的矩阵,问有多少种情况满足每一列取一个某个数字,加起来为0.

【解题思路】 Hash.

【AC代码】

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
#define CLE(a,b) memset(a,b,sizeof(a))
#define MEC(a,b) memcpy(a,b,sizeof(a))
const int mod = 16000000;//6-8 digit prime.
int has[16000010];
int cnt[16000010];
int n,a[4010][4];
void make_hash(int num)
{
    int key = (num%mod+mod)%mod;//divisible hash.
    while((has[key]!=num)&&cnt[key]!=0) key = (key+1)%mod;//liner detect.
    has[key] = num;
    cnt[key] ++;
}
int get_num(int num)
{
    int key = (num%mod+mod)%mod;
    while((has[key]!=num)&&cnt[key]!=0) key = (key+1)%mod;
    return cnt[key];
}

int main()
{
    while(~scanf("%d",&n))
    {
        for(int i=1; i<=n; i++)
        {
            for(int j=0; j<4; j++)
            {
                scanf("%d",&a[i][j]);
            }
        }
        for(int i=1; i<=n; i++)
        {
            for(int j=1; j<=n; j++)
            {
                make_hash(a[i][0]+a[j][1]);
            }
        }
        int ans = 0;
        for(int i=1; i<=n; i++)
        {
            for(int j=1; j<=n; j++)
            {
                ans += get_num(-(a[i][2]+a[j][3]));
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}

你可能感兴趣的:(POJ 2785 4 Values whose Sum is 0)