UVa10269 Adventure of Super Mario

        题意:有A个村庄,B个城堡,给出M条路和长度。超级马里奥和公主要从最后一个城堡走到第一个村庄,单位时间走单位距离。他有一个能瞬移的玩意,可以在村庄/城堡开始/停止瞬移,瞬移到城堡必须停,最大瞬移距离为L,最多瞬移K次。求最短时间。

        思路:搜索。开一个数组vis[i][j][k],保存的是时间,i是节点编号,j是剩余瞬移次数,k是当前瞬移已经瞬移的距离,搜索中遇到比原vis[i][j][k]更优的值才往下深入。这种题在之前我看来是不能做的,现在遇到这种题居然把它A掉了!!!


#include <iostream>
#include <stdio.h>
#include <cmath>
#include <algorithm>
#include <iomanip>
#include <cstdlib>
#include <string>
#include <memory.h>
#include <vector>
#include <queue>
#include <stack>
#include <ctype.h>
#define INF 1000000
using namespace std;

struct edge{
	int v;
	int w;
};

struct node{
	int u;
	int t;//已花时间 
	int dis;//本次飞的距离 
	int k;//剩余次数 

	node(int p0,int p1,int p2,int p3){u=p0;t=p1;dis=p2;k=p3;}
};

int vis[110][15][1010];

int main(){
	int T;
	cin>>T;
	
	int A,B,M,L,K;
	
	while(T--){
		vector<edge> E[110];
		cin>>A>>B>>M>>L>>K;
		for(int i=1;i<=A+B;i++){
			for(int j=0;j<=K;j++)
				for(int k=0;k<=L;k++)vis[i][j][k]=INF;
		}
		int u,v,w;
		for(int i=1;i<=M;i++){
			cin>>u>>v>>w;
			edge e;e.v=v;e.w=w;
			E[u].push_back(e);
			edge e2;e2.v=u;e2.w=w;
			E[v].push_back(e2);
		}
		
		int ans=INF;
		queue<node> que;
		node start(A+B,0,0,K);
		vis[A+B][K][0]=0;
		que.push(start);
		while(!que.empty()){
			node cur=que.front();que.pop();
			for(int i=0;i<E[cur.u].size();i++){
				//走 
				node newn(E[cur.u][i].v,cur.t+E[cur.u][i].w,0,cur.k);
				if( newn.t<vis[newn.u][newn.k][newn.dis] ){
					que.push(newn);
					vis[newn.u][newn.k][newn.dis]=newn.t;
					if(newn.u==1&&newn.t<ans)ans=newn.t;
				}
				//开始瞬移 
				if(E[cur.u][i].w<=L&&cur.k){
					node newn(E[cur.u][i].v,cur.t,E[cur.u][i].w,cur.k-1);
					if( newn.t<vis[newn.u][newn.k][newn.dis] ){
						que.push(newn);
						vis[newn.u][newn.k][newn.dis]=newn.t;
						if(newn.u==1&&newn.t<ans)ans=newn.t;
					}
				}
				//继续瞬移 
				if(cur.dis!=0){
					if(cur.u<=A&&cur.dis+E[cur.u][i].w<=L){
						node newn(E[cur.u][i].v,cur.t,cur.dis+E[cur.u][i].w,cur.k);
						if( newn.t<vis[newn.u][newn.k][newn.dis] ){
							que.push(newn);
							vis[newn.u][newn.k][newn.dis]=newn.t;
							if(newn.u==1&&newn.t<ans)ans=newn.t;
						}
					}
				}
			}
		}
		cout<<ans<<endl;
	}
	return 0;
}


你可能感兴趣的:(搜索,uva)