题目
思路
暴搜显然会TLE,所以这时候就应该请出DFS的伙伴——折半搜索(meet in the middle)了
折半搜索的思路就是先搜完后一半后,借助这一半的数据来搜索前一半,效率是原来的2倍
这个题怎么才能折半搜索呢?
看这个方程
我们可以将其拆成两半:
设k为n/2
这样就可以进行折半搜索了
注:在搜后一半的时候需要存的是负的值
代码
#include<bits/stdc++.h>
using namespace std;
const int N=45;
int n,m,k[N],p[N];
int b[1<<23],tot;
long long ans;
void dfs_r(int i,int sum){
if(i==n) { b[++tot]=-sum;return; }
for(int x=1;x<=m;x++){
int t=k[i];
for(int j=1;j<=p[i];j++) t*=x;
dfs_r(i+1,sum+t);
}
}
void dfs_l(int i,int sum){
if(i==n/2) { int x=sum;ans+=upper_bound(b+1,b+1+tot,x)-lower_bound(b+1,b+1+tot,x);return; }//有多少个相同的解
for(int x=1;x<=m;x++){
int t=k[i];
for(int j=1;j<=p[i];j++) t*=x;
dfs_l(i+1,sum+t);
}
}
int main()
{
ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
cin>>n>>m;
for(int i=0;i<n;i++) cin>>k[i]>>p[i];
dfs_r(n/2,0);
sort(b+1,b+1+tot);
dfs_l(0,0);
cout<<ans<<endl;
return 0;
}
end
完结撒花