【SSL】1127&【POJ】1186&【洛谷】P5691方程的解数(HASH+折半搜索)
Time Limit:1000MS
Memory Limit:65536K
Description
Input
第1行包含一个整数n。第2行包含一个整数M。第3行到第n+2行,每行包含两个整数,分别表示ki和pi。两个整数之间用一个空格隔开。第3行的数据对应i=1,第n+2行的数据对应i=n。
Output
仅一行,包含一个整数,表示方程的整数解的个数。
Sample Input
3
150
1 2
-1 2
1 2
Sample Output
178
思路
用折半搜索,
先搜前一半,存入hash表。
再搜后一半,在hash表中找。
用一个数组保存次数。
代码
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
long long n,m,k[10],p[10],a[10],b[151][10],ans=0,hs[4000007],cs[4000007],len=4000007;
long long HASH(long long x)//hs函数
{
for(;x<0;x+=4000007);
return x%4000007;
}
int locate(long long x)//x定位
{
long long i,w=HASH(x);
for(i=0;i<len&&hs[(w+i)%len]!=0&&hs[(w+i)%len]!=x;i++);
return (w+i)%len;
}
void insert(long long x)//插入x
{
long long w=locate(x);
hs[w]=x,cs[w]++;
return;
}
void DFS(long long dep)//折半搜索
{
long long i,t;
if(dep>n/2)
{
for(t=0,i=1;i<dep;i++)
t+=k[i]*b[a[i]][p[i]];
insert(-t);
return;
}
for(i=1;i<=m;i++)
{
a[dep]=i;
DFS(dep+1);
}
return;
}
void dfs(long long dep)//折半搜索
{
long long i,t,w;
if(dep>n)
{
for(t=0,i=n/2+1;i<=n;i++)
t+=k[i]*b[a[i]][p[i]];
w=locate(t);
if(hs[w]==t)
ans+=cs[w];
return;
}
for(i=1;i<=m;i++)
{
a[dep]=i;
dfs(dep+1);
}
return;
}
void input()
{
long long i,j,l,x;
scanf("%lld%lld",&n,&m);
for(i=1;i<=n;i++)
scanf("%lld%lld",&k[i],&p[i]);
for(i=1;i<=m;i++)
for(j=1;j<=n;j++)
for(b[i][p[j]]=1,l=0;l<p[j];b[i][p[j]]*=i,l++);
return;
}
int main()
{
input();
DFS(1);
dfs(n/2+1);
printf("%lld",ans);
return 0;
}