POJ2331Water pipe(IDA*)

最新推荐文章于 2020-01-14 10:53:01 发布

帐下幕僚

最新推荐文章于 2020-01-14 10:53:01 发布

阅读量326

点赞数

分类专栏： acm之路 dfs/bfs 思路++

本文链接：https://blog.csdn.net/abc13068938939/article/details/52196468

版权

acm之路同时被 3 个专栏收录

107 篇文章 0 订阅

订阅专栏

思路++

31 篇文章 0 订阅

订阅专栏

dfs/bfs

9 篇文章 0 订阅

订阅专栏

本来用哈弗曼距离作为h函数，T了。看了大神的博客才知道要先对x轴，y轴预处理h函数。也就是对于终点的ed.x，要在0< x< = 1000的范围中将h(x)先算出来（h(ed.x)=0）,h[tp+L[i]]=h[tp]+1;h[tp-L[i]]=h[tp]+1。

#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <stdlib.h>
#include <stack>
#include <vector>
#include <string.h>
#include <queue>
#define msc(X) memset(X,-1,sizeof(X))
#define ms(X) memset(X,0,sizeof(X))
#define mabs(X) ((X)>0?(X):(-(X)))
typedef long long LL;
using namespace std;
int L[6],C[6],k,mln=0;
struct _Point
{
    int x,y;
    int C[6];
}st,ed;
void cal_h(int *h,int t)
{
    queue<int > q;
    while(!q.empty()) q.pop();
    h[t]=0;
    q.push(t);
    while(!q.empty()){
        int tp=q.front();
        q.pop();
        for(int i=0;i<k;i++)
        {       
            if(tp+L[i]<=1000&&h[tp+L[i]]==-1){
                h[tp+L[i]]=h[tp]+1;
                q.push(tp+L[i]);
            }
            if(tp-L[i]>0&&h[tp-L[i]]==-1){
                h[tp-L[i]]=h[tp]+1;
                q.push(tp-L[i]);
            }
        }
    }
}
int pLt,hx[1020],hy[1020];
bool IDA(struct _Point cur,int d,int type)
{
    int hv=type?hy[cur.y]:hx[cur.x];
    if(hv==-1||hv+d>pLt) return false;
    if(hv==0){
        if(type) {printf("%d\n",pLt );
            return true;}
        else return IDA(cur,d,1);
    }
    struct _Point next;
    for(int j=0;j<k;j++)
    {
        if(cur.C[j]>=C[j]) continue;
        if(!type){
            next=cur;
            next.C[j]++;
            next.x+=L[j];
            if(next.x<=1000&&IDA(next,d+1,type))
                return true;
            next=cur;
            next.C[j]++;
            next.x-=L[j];
            if(next.x>0&&IDA(next,d+1,type))
                return true;
        }
        else{
            next=cur;
            next.C[j]++;
            next.y+=L[j];
            if(next.y<=1000&&IDA(next,d+1,type))
                return true;
            next=cur;
            next.C[j]++;
            next.y-=L[j];
            if(next.y>0&&IDA(next,d+1,type))
                return true;
        }
    }
    return false;
}
int main(int argc, char const *argv[])
{
    int mx=0;
    scanf("%d %d %d %d",&st.x,&st.y,&ed.x,&ed.y);
    scanf("%d",&k);
    for(int i=0;i<k;i++) {scanf("%d",L+i);mln=max(mln,L[i]);}
    for(int i=0;i<k;i++) {scanf("%d",C+i);mx+=C[i];}
    for(int i=0;i<k;i++)
        st.C[i]=ed.C[i]=0;
    msc(hx);
    msc(hy);
    cal_h(hx,ed.x);
    cal_h(hy,ed.y);
    if(hx[st.x]==-1||hy[st.y]==-1) {puts("-1");return 0;}
    for(pLt=hx[st.x]+hy[st.y];pLt<=mx;pLt++)
        if(IDA(st,0,0)) break;
    if(pLt>mx) puts("-1");
    return 0;
}