code vs 1735 方程的解数 （dfs+hash）

题目描述 Description

已知一个n元高次方程：

k1x1p1+k2x2p2+……+knxnpn = 0

其中：x1, x2, …,xn是未知数，k1,k2,…,kn是系数，p1,p2,…pn是指数。且方程中的所有数均为整数。

假设未知数1≤ xi ≤M, i=1,,,n，求这个方程的整数解的个数。

输入描述 Input Description

文件的第1行包含一个整数n。第2行包含一个整数M。第3行到第n+2行，每行包含两个整数，分别表示ki和pi。两个整数之间用一个空格隔开。第3行的数据对应i=1，第n+2行的数据对应i=n。

输出描述 Output Description

文件仅一行，包含一个整数，表示方程的整数解的个数。

样例输入 Sample Input

3

150

1  2

-1  2

1  2

样例输出 Sample Output

178

数据范围及提示 Data Size & Hint

 1≤n≤6；1≤M≤150；

|k1Mp1|+|k2Mp2|+……+|knMpn |< 231

方程的整数解的个数小于231。

★本题中，指数Pi(i=1,2,……,n)均为正整数。

题解：dfs+hash

折半搜索，对于式子中的前一半爆搜出所有的值存入hash表中。

然后在搜索剩余一半的答案，判断hash表中是否有能组合形成0的数，计算即可。

#include<iostream>  
#include<cstring>  
#include<cstdio>  
#include<algorithm>  
#include<cmath>  
#define N 100  
#define mod 7231773
using namespace std;  
struct data{
	int val,num;
}mp[mod+10];
int n,m,cnt,t;  
int k[N],p[N],mi[160][10];  
int hash(int val)
{
	int c=abs(val)%mod;
	while (mp[c].num>0&&mp[c].val!=val) c=(c+1)%mod;
	return c;
}
void dfs(int x,int sum)
{
	if (x>t) {
	   int pos=hash(sum);
       mp[pos].val=sum; mp[pos].num++; 
	   return;
    }
	for (int i=1;i<=m;i++) 
      dfs(x+1,sum+mi[i][p[x]]*k[x]);
}
void dfs1(int x,int sum)
{
	if (x>n) {
	  int pos=hash(-sum);
	  cnt+=mp[pos].num;
	  return;
    }
	for (int i=1;i<=m;i++) dfs1(x+1,sum+mi[i][p[x]]*k[x]);
}
int main()  
{  
    freopen("equation1.in","r",stdin);  
    //freopen("equation1.out","w",stdout);  
    scanf("%d%d",&n,&m);  
    for (int i=1;i<=n;i++) scanf("%d%d",&k[i],&p[i]);  
    for (int i=1;i<=m;i++) {  
        mi[i][0]=1;  
        for (int j=1;j<=6;j++) mi[i][j]=mi[i][j-1]*i;  
    }  
    if (n==1) {  
        int ans=0;
        for (int i=1;i<=m;i++) {  
            ans=mi[i][p[1]]*k[1]; 
			//cout<<mi[i][p[1]]<<" "<<ans<<endl; 
            if (!ans) cnt++;  
        }  
        printf("%d\n",cnt);  
        return 0;  
    }  
    t=n/2;  dfs(1,0);
    dfs1(t+1,0);
    printf("%d\n",cnt);  
}