poj1840Eqs（hash)

http://poj.org/problem?id=1840

枚举 xi不等0.

#include <iostream>
#include<cstdio>
#include<cstring>
using namespace std;
#define MOD 97777
#define size 1000010
int t,head[size],next[size],v[size],num[size][2];
void init()
{
    t =0 ;
    memset(head,-1,sizeof(head));
}
void add(int a,int b)
{
    next[t] = head[a];
    head[a] = t;
    v[t] = b;
    t++;
}
int main()
{
    int i,j,k,a[6],s,g=0;
    init();
    for(i = 1; i <= 5 ; i++)
    cin>>a[i];
    for(i = -50; i <= 50 ;i++)
    {
        if(i==0)
        continue;
        for(j = -50; j <= 50 ; j++)
        {
            g++;
            if(j==0)
            continue;
            s = a[1]*(i*i*i)+a[2]*(j*j*j);
            s = s%MOD;
            if(s<0)
            s+=MOD;
            num[g][0] = i;
            num[g][1] = j;
            add(s,g);
        }
    }
    int sum = 0,o,ss;
    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;
                s = a[3]*(i*i*i)+a[4]*(j*j*j)+a[5]*(k*k*k);
                ss = s;
                s = (-s)%MOD;
                if(s<0)
                s += MOD;
                for(o = head[s] ; o!=-1 ; o = next[o])
                {
                    int i1 = num[v[o]][0],i2 = num[v[o]][1];
                    if((a[1]*(i1*i1*i1)+a[2]*(i2*i2*i2))==-ss)
                         sum++;
                }
            }
        }
    }
    cout<<sum<<endl;
    return 0;
}