HDU1496Equations【hash】

 1 #include <iostream>
 2 #include <cstdio>
 3 #include <cstring>
 4 using namespace std;
 5 
 6 const int mod = 40007;
 7 
 8 struct Node {
 9     int to;
10     int next;
11 }e[mod + 10];
12 
13 int head[mod + 10];
14 
15 int tot;
16 
17 void Add(int u, int v) {
18     e[tot].to = v;
19     e[tot].next = head[u];
20     head[u] = tot++;
21 }
22 
23 int Find(int u, int v) {
24     int cnt = 0;
25     for(int i = head[u]; i; i = e[i].next) {
26         if(e[i].to == v) cnt++;
27     }
28     return cnt;
29 }
30 
31 void init() {
32     memset(head, 0, sizeof(head));
33     tot = 1;
34 }
35 
36 int Fabs(int x) {
37     return x > 0 ? x : - x;
38 }
39 
40 int pre[105];
41 
42 int main() {
43     int a, b, c, d;
44     for(int i = 1; i <= 100; i++) {
45         pre[i] = i * i;
46     }
47     while(EOF != scanf("%d %d %d %d",&a, &b, &c, &d) ) {
48         init();
49         for(int i = 1; i <= 100; i++) {
50             for(int j = 1; j <= 100;    j++) {
51                 int num = a * pre[i] + b * pre[j];
52                 int p = Fabs(num) % mod;
53                 Add(p, num);
54             }
55         } 
56         int ans = 0;
57         for(int i = 1; i <= 100; i++) {
58             for(int j = 1; j <= 100;    j++) {
59                 int num = c * pre[i] + d * pre[j];
60                 int p = Fabs(num) % mod;
61                 ans += Find(p, - num);
62             }
63         } 
64         printf("%d\n", ans << 4);
65     }
66     return 0;
67 }