等式 hash

刚开始想都不用想用了暴力 当然超时。

根据题目要求10000k内存 

一般64m内存即可以通过

利用hash函数建立，时间通过。但内存超出了。。。代码1

        
        
        #include <cstdio>
#include <cstring>
using namespace std;

short hash1[25000010];      //存放hash函数的对应关系的数据

int main()
{
    int a[5],t;
    int temp;
    long long sum;
    scanf("%d",&t);
    while(t--)
    {
        sum=0;
        memset(hash1,0,sizeof(hash1));
        for(int i=0;i<5;i++)
            scanf("%d",a+i);
        for(int i=-50;i<=50;i++)
        {
            if(i==0)
                continue;
            for(int j=-50;j<=50;j++)
            {
                if(j==0)    continue;
                temp=a[0]*i*i*i+a[1]*j*j*j;
                if(temp<0)
                    temp+=25000001;
                hash1[temp]++;
            }
        }
        for(int i=-50;i<=50;i++)
        {
            if(i==0)    continue;
            for(int j=-50;j<=50;j++)
            {
                if(j==0)    continue;
                for(int z=-50;z<=50;z++)
                {
                    if(z==0)    continue;
                    temp = -(a[2]*i*i*i+a[3]*j*j*j+a[4]*z*z*z);
                    if(temp<0)
                        temp+=25000001;
                    if(temp>=0&&temp<25000001)
                        sum+=hash1[temp];
                }
            }
        }
        printf("%d
",sum);
    }
    return 0;
}
        
        
        

处理冲突使用链地址法代码2

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<string.h>
#include<stdlib.h>
#include<algorithm>
#include<map>
using namespace std;
const int p = 19373;
const int a = 2* 50*50*50*50;
typedef struct Node
{
    int value,num;
    struct Node * next;
}node;
node * res[p];//结构体指针数组
int main()
{
    int T,x1,x2,x3,x4,x5,a1,a2,a3,a4,a5;
    int i ,j,k;
    //node* res[p];
    for(i = 0; i < p; i++)
    res[i] = new node;//创建哈希点
    while(cin>>T)
    {
        while(T--)
        {
            int t ,x;
            for(i = 0; i< p; i++)//哈希点初始化
            {
                res[i]->next = NULL;
                res[i]->num = 0;
            }
            cin>>a1>>a2>>a3>>a4>>a5;
            for(i = -50;i <= 50 ; i++)//枚举x1，x2
            {
                if(i == 0)continue;//注意 所有的x 不能为0
                for(j = -50; j <= 50; j++)
                {
                    if(j == 0) continue;
                    x = a1*i*i*i + a2*j*j*j;
                    int sum = x >= 0?x:-x;//取模要用绝对值
                    sum %= p;
                    node* p1;//这个地方注意不要开成节点了，下面会让它指向哈希点下一个位置，指针就行了，不然会超内存
                    node* p2 = new node;//
                    p1 = res[sum]->next;
                    int flag = 0;
                    while(p1)//沿着哈希点找，如果找到就标记flag  = 1，找不到就把这个数插进去
                    {
                        if(p1->value == x)
                        {
                            flag = 1;
                            p1->num++;
                            delete p2;
                            break;
                        }
                        p1 = p1->next;
                    }
                    if(!flag)//没有找到，用p2  保存这个数的信息，插入到哈希表中
                    {
                        p2->value = x;
                        p2->num= 1;
                        p2->next = res[sum]->next;
                        res[sum]->next = p2;
                    }
                }
            }
            long long ans =0;
            for(i = -50; i <= 50; i++)
            {
                if(i == 0) continue;
                for(j = -50; j <= 50; j++)
                {
                    if(j == 0) continue;
                    for(k = -50; k <= 50; k++)
                    {
                        if(k == 0) continue;
                        x = a3*i*i*i + a4*j*j*j + a5*k*k*k;
                        int sum = x >0? x:-x;
                        if(sum > a) continue;//优化的地方，如果大于 a 就不用找了
                        sum %= p;

                        node* p1 = res[sum]->next;
                        while(p1)
                        {
                            if(p1->value == -x)
                            {
                                ans += p1->num;
                                break;
                            }
                            else
                            p1 = p1->next;
                        }

                    }
                }
            }
            cout<<ans<<endl;
        }
    }
    return 0;
}

最优代码：

 
#include<iostream>
#include<algorithm>
#include<cstdio>
#define N 101

using namespace std;

int A[N*N],B[N*N],C[N],len1,len2,len3;      

bool cmp(int x,int y)
{
    if(x>y) return true;
    return false;
}

void work()
{
    int i,j,k,key;
    sort(A,A+len1);
    sort(B,B+len2,cmp);
    long long ans=0;
    for(k=0;k<len3;k++)
    {
        key=-C[k];
        i=0;j=0;
        while(i<len1&&j<len2)
        {
            if(key>A[i]+B[j]) i++;
            else if(key<A[i]+B[j]) j++;
            else
            {
                int p,q;
                p=q=1;
                while(A[i]==A[i+1]&&i+1<len1){p++;i++;}
                while(B[j]==B[j+1]&&j+1<len2){q++;j++;}
                ans+=p*q;
                i++;j++;
            }
        }
    }
    cout<<ans<<endl;
}

int main()
{
    int test,a1,a2,a3,a4,a5,x1,x2,x3,x4,x5;
    scanf("%d",&test);
    while(test--)
    {
        scanf("%d%d%d%d%d",&a1,&a2,&a3,&a4,&a5);
        len1=0;
        for(x1=-50;x1<=50;x1++)
        {
            if(x1==0) continue;
            for(x2=-50;x2<=50;x2++)
            {
                if(x2==0) continue;
                A[len1++]=a1*x1*x1*x1+a2*x2*x2*x2;
            }
        }
        len2=0;
        for(x3=-50;x3<=50;x3++)
        {
            if(x3==0) continue;
            for(x4=-50;x4<=50;x4++)
            {
                if(x4==0) continue;
                B[len2++]=a3*x3*x3*x3+a4*x4*x4*x4;
            }
        }
        len3=0;
        for(x5=-50;x5<=50;x5++)
        {
            if(x5==0) continue;
            C[len3++]=a5*x5*x5*x5;
        }
        work();
    }
    return 0;
}