[NOI2001]方程的解数

题目
DFS(meet in the middle)
这里写图片描述
通过此条件剪枝,由于本题数据较大,可通过HASH表存储方案数。

#include<cstdio>
#include<vector>
#include<cmath>
#include<cstdlib>
using namespace std;
#define MAXN 6
#define MAXH 4999999
struct node{
    int pos,cnt;
    node *next;
}*adj[MAXH+10];
void addnode(int pos){
    node *p=new node;
    p->pos=pos;
    p->cnt=1;
    p->next=adj[abs(pos%MAXH)];
    adj[abs(pos%MAXH)]=p;
}
int n,m,k[MAXN+2],p[MAXN+2],mid,ans,mypow[160][10];
void read(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
        scanf("%d%d",&k[i],&p[i]);
}
void dfs1(int i,int sum){
    if(i>mid){
        int h=abs(sum%MAXH);
        node *p;
        for(p=adj[h];p;p=p->next)
            if(p->pos==sum){
                p->cnt++;
                break;
            }
        if(!p)
            addnode(sum);
        return;
    }
    for(int x=1;x<=m;x++){
        long long t=sum+k[i]*(long long)mypow[x][i];
        if(abs(t)>0x7f7f7f7f)
            break;
        dfs1(i+1,(int)t);
    }
}
void dfs2(int i,int sum){
    if(i==mid){
        int h=abs(sum%MAXH);
        node *p;
        for(p=adj[h];p;p=p->next)
            if(p->pos==sum){
                ans+=p->cnt;
                break;
            }
        return;
    }

    for(int x=1;x<=m;x++){
        long long t=sum-k[i]*(long long)mypow[x][i];
        if(abs(t)>0x7f7f7f7f)
            break;
        dfs2(i-1,(int)t);
    }
}
int main()
{

    read();
    int i,j;
    for(i=1;i<=150;i++){
        for(j=1;j<=n;j++)
            mypow[i][j]=pow(i,p[j]);
    }
    mid=n/2;
    dfs1(1,0);
    dfs2(n,0);
    printf("%d\n",ans);
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值