pku 1186(DFS+hash)

将方程一分为二，一半元素留在左边，另一边移至右边，然后分2次DFS，搜索左边的元素时，将每次取到的总和存入hash。

若重复出现，则增加引用计数。

DFS右边时，对每次总和在hash中查找，若找到该元素，则sum加上该元素的出现次数。注意右边的总和要取反。

最终sum即为解的个数。

 

#include <iostream>#include <vector>using namespace std;class node{public:int cnt,value;};vector<node> hash[100005];int k[6],p[6];int n,m,mid,sum=0;int Find(int x,int s)//查找函数{int sz=hash[x].size();for(int i=0;i<sz;++i){if(hash[x][i].value==s)return i;}return -1;}void DFS_left(int x,int s){if(x>=mid){bool f;//这个用来处理s小于0的情况if(s>=0) f=true;else{s= -s;f=false;}int t=s%100003,tag;if(!f) s= -s;if( (tag=Find(t,s)) != -1)//若表中已存在该元素，引用次数加1{hash[t][tag].cnt++;}else {node tmp;tmp.cnt=1;tmp.value=s;hash[t].push_back(tmp);//压入新元素}return;}for(int i=1;i<=m;++i)//枚举{int t=1;for(int j=0;j<p[x];++j)t *= i;t*=k[x];DFS_left(x+1,s+t);}}void DFS_right(int x,int s){if(x>=n){s= -s;//移至右边的元素要取反bool f;if(s>=0) f=true;else {s= -s;f=false;}int t=s%100003,tag;if(!f) s= -s;if( (tag=Find(t,s)) != -1)//若找到该元素，则解得个数加上该元素出现次数{sum += (hash[t][tag].cnt);}return;}for(int i=1;i<=m;++i)//枚举{int t=1;for(int j=0;j<p[x];++j)t *= i;t=t*k[x];DFS_right(x+1,s+t);}}int main(){cin>>n>>m;mid=n/2;for(int i=0;i<n;++i)cin>>k[i]>>p[i];DFS_left(0,0);DFS_right(mid,0);cout<<sum<<endl;return 0;}