【SSL】1127&【POJ】1186&【洛谷】P5691方程的解数(HASH+折半搜索)

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

猜你喜欢

转载自blog.csdn.net/weixin_46975572/article/details/112970352
今日推荐