Hdu 1496 hash

题目链接

对于方程 a*x1^2 + b*x2^2 + c*x3^2 + d*x4^2 = 0　　(　-50<=a, b, c, d <=50, 且均不为０，-100 <= x1, x2, x3, x4 <= 100，　且均不为０）

在已知a, b, c, d的情况下，计算有多少组x1, x2, x3, x4满足要求．

如果枚举x1, x2, x3，　那么理论上10^6可解，　不过我没有尝试．．．

采用另一种思路，　a*x1^2 + b*x2^2 = -c*x3^2 + (-d)*x4^2

等式两边范围均为：-100000０~1000000

附上代码：

/*************************************************************************
    > File Name: 1496.cpp
    > Author: Stomach_ache
    > Mail: sudaweitong@gmail.com
    > Created Time: 2014年05月14日 星期三 16时22分46秒
    > Propose: 
 ************************************************************************/

#include <cmath>
#include <string>
#include <cstdio>
#include <vector>
#include <fstream>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;

int a, b, c, d;
#define MAX_N (1000000+2)
int hash[MAX_N<<1];
//标记每次hash值改变的下标
//若每次都用memset(hash, 0, sizeof(hash))会ＴＬＥ
vector<int> flag;

int
main(void) {
    memset(hash, 0, sizeof(hash));
      while (~scanf("%d %d %d %d", &a, &b, &c, &d)) {
          c *= -1;
        d *= -1;
        for (int i = 1; i <= 100; i++) {
              for (int j = 1; j <= 100; j++) {
                  hash[a*i*i+b*j*j+MAX_N] += 4;
                flag.push_back(a*i*i+b*j*j+MAX_N);
            }
        }
        int ans = 0;
        for (int i = 1; i <= 100; i++) {
              for (int j = 1; j <= 100; j++) {
                  if (hash[c*i*i+d*j*j+MAX_N] > 0) {
                      ans += 4 * hash[c*i*i+d*j*j+MAX_N];
                }
            }
        }
        for (size_t i = 0; i < flag.size(); i++) {
              hash[flag[i]] = 0;
        }
        flag.clear();
        
        printf("%d
", ans);
    }

    return 0;
}