1840题Hash解法

1、题目描述

Eqs

Time Limit: 5000MS		Memory Limit: 65536K
Total Submissions: 5535		Accepted: 2662

Description

Consider equations having the following form:
a1x1 ³ + a2x2 ³ + a3x3 ³ + a4x4 ³ + a5x5 ³ =0
The coefficients are given integers from the interval [-50,50].
It is consider a solution a system (x1, x2, x3, x4, x5) that verifies the equation, xi∈[-50,50], xi != 0, any i∈{1,2,3,4,5}.

Determine how many solutions satisfy the given equation.

Input

The only line of input contains the 5 coefficients a1, a2, a3, a4, a5, separated by blanks.

Output

The output will contain on the first line the number of the solutions for the given equation.

Sample Input

37 29 41 43 47

Sample Output

654

Source

Romania OI 2002

2、题意解读

     此题如果直接枚举的话，有100^5次枚举，肯定超时。

可以作如下变换：                      

-(a1x1³ + a2x2³ ) = a3x3³ + a4x4³ + a5x5³

    左边的数据最多有100^2，可以以此建立Hash表，然后在Hash表中，对右边100^3的数据中进行枚举查找，这样就将复杂度从O(100^5)降为O(100^3)。

    建立Hash表过程注意以下几点：

（1）插入过程：关键字key设为-(a1x13+ a2x23)，相同的关键字情况下，要注意统计重复次数。因为虽然key相同，但是它对应着不同的x1、x2，即不同的解。

（2）解决碰撞：可以用拉链法或者开放寻址法解决碰撞，拉链法可以使用指针链表或者静态链表。从实测的效果看，开发寻址法，超时，指针链表有时可以过，有时超时；静态链表效果最好，可以达到300多MS。

3、C代码：

（1）指针链表法的关键代码：

#define H_MAX 19997 typedef struct __Node{ int key; int count; struct __Node *next; }Node; void insert(Node HashTable[H_MAX], int k){ int hash_key = abs(k)%H_MAX; Node *newNode = NULL; Node *p = HashTable[hash_key].next; while(p != NULL){ if(p->key == k){ p->count++; return; }else{ p = p->next; } } newNode = (Node*)malloc(sizeof(Node)); if(newNode == NULL) return; newNode->key = k; newNode->count = 1; newNode->next = HashTable[hash_key].next; HashTable[hash_key].next = newNode; } int search(Node HashTable[H_MAX], int k){ int hash_key = abs(k)%H_MAX; Node *p = HashTable[hash_key].next; while(p != NULL){ if(p->key == k) return p->count; p = p->next; } return 0; }

（4）静态链表法完整代码

#include <stdio.h> #include <stdlib.h> #include <math.h> #define H_MAX 19997 //全局变量，保存next数组中的实际元素个数 int m=0; typedef struct __Node{ int key; //关键字 int count; //重复次数 int ne; //指向相同槽位中的下一个结点在数组next[]下标 }Node; /*由于左边方程建立的Hash表的关键字最多为10000，建立一个略大于10000的数组， *即使在最坏情况下，即所有的关键字都落在一个槽位中， *也可以保存所有结点 */ Node next[10005] = {0}; void insert(Node HashTable[H_MAX], int k){ int hash_key = abs(k)%H_MAX; int i=0; //先查询 for(i=HashTable[hash_key].ne;i != -1;i=next[i].ne){ //查询成功 if(next[i].key == k){ next[i].count++; return; } } //查询失败，插入 next[m].key = k; next[m].count = 1; next[m].ne = HashTable[hash_key].ne; HashTable[hash_key].ne = m++; } int search(Node HashTable[H_MAX], int k){ int hash_key = abs(k)%H_MAX; int i = 0; for(i=HashTable[hash_key].ne; i!=-1; i=next[i].ne){ if(next[i].key == k) return next[i].count; } return 0; } int main(int argc, char **args){ int a1,a2,a3,a4,a5; int sum = 0; int i,j,k; int max = 0, min = 0; int key = 0; Node HashTable[H_MAX] = {0}; scanf("%d%d%d%d%d", &a1, &a2, &a3, &a4, &a5); max = (int)( (abs(a1)+abs(a2)) * 50*50*50 ); min = -max; for(i=0;i<H_MAX;++i){ HashTable[i].key = 0; HashTable[i].count = 0; HashTable[i].ne = -1; } for(i=-50;i<=50;++i){ if(i == 0) continue; for(j=-50;j<=50;++j){ if(j == 0) continue; key = -(int)( a1*i*i*i+a2*j*j*j ); insert(HashTable, key); } } sum = 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; key = (int)(a3*i*i*i+a4*j*j*j+a5*k*k*k); //简单的剪枝 if(key>max || key <min){ continue; } else{ sum += search(HashTable, key); } } } } printf("%d/n", sum); return 0; }