POJ 2331 Water pipe IDA*

预处理估价函数。其他的套模板就可以了。

不过有一个地方没住意WA了2小时。

#include<stdio.h>
#include<string.h>
#include<queue>
using namespace std;
int sx,sy,tx,ty,n;
int len[5],num[5];
int bound,total;
int h[2]={1,-1};
int hx[1010],hy[1010];
bool ans;
struct point{
	int x;
	int step;
};
int Min(int a,int b){
	return a>b?b:a;
}

void bfs(int v,int * hx){
	int i,j,u,xx;
	queue<struct point>que;
	struct point sta;
	sta.step=0;
	sta.x=v;
	que.push(sta);
	hx[v]=0;
	while(!que.empty()){
		struct point tem;
		tem=que.front();
		que.pop();
		u=tem.x;
		for(i=0;i<2;i++){
			for(j=1;j<=n;j++){
					xx=u+len[j]*h[i];
					if(xx<=0 || xx>1000 || hx[xx]!=-1)
						continue;
					if(hx[u]+1>total)
						continue;
					hx[xx]=hx[u]+1;
					struct point cur;
					cur.step=hx[xx];
					cur.x=xx;
					que.push(cur);
				}

		}

	}
}

int dfs(int dep,int x,int *nu,int type){
	int i,j,xx,next_bound;
	if(type==0){
		if(dep+hx[x]+hy[sy]>bound)
			return dep+hx[x]+hy[sy];
		if(hx[x]==0)
			return dfs(dep,sy,nu,1);
	}
	else if(type==1){
		if(dep+hy[x]>bound)
			return dep+hy[x];
		if(hy[x]==0){
			ans=true;
			return dep;
		}
	}
	int tmp[5];  
	for(i=1;i<=n;i++)  
		tmp[i]= nu[i];  

	int new_bound=1e7;
	for(i=0;i<2;i++){
		for(j=1;j<=n;j++)
			if(tmp[j]){
				tmp[j]--;
				xx=x+h[i]*len[j];
				if(xx<=0 || xx>1000 ){
                    tmp[j]++;                   //一开始我这里忘了去恢复现场，WA了好久
					continue;
                }
				if(type==0){
					if(hx[xx]==-1){
                            tmp[j]++;
					        continue;           //一开始我这里忘了去恢复现场，WA了好久
                    }
                }
				if(type==1){
					if(hy[xx]==-1){
                            tmp[j]++;
					        continue;           //一开始我这里忘了去恢复现场，WA了好久
                     }
				}
				next_bound = dfs(dep+1,xx,tmp,type);
				if(!ans)
					new_bound=Min(new_bound,next_bound);
				else
					return next_bound;
				tmp[j]++;
			}
	}
	return new_bound;
}

void IDA_STAR(){
	
	memset(hx,-1,sizeof(hx));  
    memset(hy,-1,sizeof(hy));  

	bfs(tx,hx);
	bfs(ty,hy);
	if(hx[sx]==-1 || hy[sy]==-1){
		printf("-1\n");
		return;
	}
	bound=hx[sx]+hy[sy];
	while(bound<=total && !ans){
		bound=dfs(0,sx,num,0);
	}
	if(ans)
		printf("%d\n",bound);
	else
		printf("-1\n");
}
int main(){
	int i,j;
	scanf("%d%d%d%d",&sx,&sy,&tx,&ty);  
	scanf("%d",&n);  
	for(i=1;i<=n;i++)
		scanf("%d",&len[i]);
	total=0;
	for(i=1;i<=n;i++){
		scanf("%d",&num[i]);
		total+=num[i];
	}
	ans=false;
	IDA_STAR();
}