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; }