SSLOJ1127 方程的解数&P5691

Description

已知一个 n 元高次方程：
$\sum _{i=1}^n{k}_i{x}_i^{p_i}=0$
其中：$x1,x2,\dots ,xn$是未知数，$k1,k2,\dots ,kn$ 是系数，$p1,p2,\dots pn$是指数。且方程中的所有数均为整数。假设未知数 $x_i\in [1,m](i\in [1,n])$，求这个方程的整数解的个数。

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

思路

直接dfs必T(zfx??,所以我们考虑折半搜索，合并左右2边可以哈希
code:

#include<cstdio>
#include<queue>
#define myd 4000000
using namespace std;
long long a[myd+80][2];
long long k[7],p[7],n,m,s,ans;
long long as(long long x)
{
 if (x<0) return -x;
 else return x;
}
long long pw(long long x,long long y)
{
 long long s=1;
 while (y--) s*=x;
 return s;
}
long long f(long long x)
{
 long long o=as(x);
 o%=myd;
 long long i=0;
 while (i<myd&&a[(o+i)][0]!=x&&a[(o+i)%myd][0]) i++;
 return (o+i)%myd;
}
bool find(long long x)
{
 return a[f(x)][0]==x;
}
void cr(long long x)
{
 a[f(x)][0]=x;
 a[f(x)][1]++;
 return;
}
void dfs(long long x)
{
 if (x==n/2+1)
 {
  cr(s);
  return;
 }
 for (long long i=1;i<=m;i++)
 {
  s+=pw(i,p[x])*k[x];
  dfs(x+1);
  s-=pw(i,p[x])*k[x];
 }
 return;
}
long long u(long long x)
{
 if (a[f(x)][0]==x) return a[f(x)][1];
 else return 0;
}
void dfs2(long long x)
{
 if (x==n+1)
 {
  ans+=u(-s);
  return;
 }
 for (long long i=1;i<=m;i++)
 {
  s+=pw(i,p[x])*k[x];
  dfs2(x+1);
  s-=pw(i,p[x])*k[x];
 }
 return;
}
int main()
{
 scanf("%lld%lld",&n,&m);
 for (long long i=1;i<=n;i++) scanf("%lld%lld",&k[i],&p[i]);
 dfs(1);
 dfs2(n/2+1);
 printf("%lld",ans);
 return 0;
}