0
点赞
收藏
分享

微信扫一扫

2022牛客寒假算法基础集训营6 G迷宫2

伽马星系 2022-02-18 阅读 44
算法图论

题目:迷宫2

思路:
对每个格子,和自己的上下左右四个格子连边,当前方向指向的边边权为0,否则边权为1。
走过一条边权为1的边相当于进行了一次修改操作。
再跑一遍最短路就可以啦。

#include<bits/stdc++.h>
using namespace std;

namespace DIJKSTRA{
	typedef long long ll;
	typedef pair<int,ll> pil;
	const int N=2e6;
	const ll inf=0x3f3f3f3f3f3f3f3f;
	
	vector<pil> g[N+5];
	ll dist[N+5];
	bool vis[N+5];
	pil fa[N+5];	//记录路径 
	
	struct cmp{
		bool operator () (pil e1,pil e2) {
			return e1.second>e2.second; 
		}
	};
	priority_queue<pil,vector<pil>,cmp> que;
	
	void init(int n,int st) {
		for(int i=1;i<=n;i++) g[i].clear(),vis[i]=0,dist[i]=inf;
		dist[st]=0;
	}
	
	void addedge(int u,int v,ll w) {
		g[u].push_back({v,w});
	}
	
	ll dijkstra(int n,int st,int ed) {
		que.push({st,0});
		while(!que.empty()) {
			int x=que.top().first;
			que.pop();
			if(vis[x]) continue;
			else vis[x]=true;
			
			for(pil t:g[x]) {
				int y=t.first;
				ll w=t.second;
				if(dist[y]>dist[x]+w) {
					dist[y]=dist[x]+w;
					que.push({y,dist[y]});
					fa[y]={x,w}; 
				}
			}
		}
		
		return dist[ed];
	}
}
using namespace DIJKSTRA;

int n,m;
string a[N+5];

int getid(int x,int y) {
	return (x-1)*m+y;
}

void slv() {
	cin>>n>>m;
	for(int i=1;i<=n;i++) {
		string s;
		cin>>s;
		a[i]=" "+s;
	}
	
	init(n*m,1);
	
	const int mv1[5]={0,-1,1,0,0};
	const int mv2[5]={0,0,0,-1,1};
	for(int i=1;i<=n;i++) {
		for(int j=1;j<=m;j++) {
			int t[5]={0,1,1,1,1};
			if(a[i][j]=='U') t[1]=0;
			else if(a[i][j]=='D') t[2]=0;
			else if(a[i][j]=='L') t[3]=0;
			else t[4]=0;
			
			for(int k=1;k<=4;k++) {
				int x=i+mv1[k],y=j+mv2[k];
				if(x>0&&y>0&&x<=n&&y<=m) addedge(getid(i,j),getid(x,y),t[k]);
			}
		}
	}
	
	cout<<dijkstra(n*m,1,n*m)<<endl;
	int t=n*m;
	while(t!=1) {
		int u=fa[t].first;
		if(fa[t].second!=0) {
			int x=(u-1)/m+1,y=((u%m==0)?m:u%m);
			cout<<x<<' '<<y<<' ';
			if(u-m==t) cout<<"U\n";
			else if(u+m==t) cout<<"D\n";
			else if(u-1==t) cout<<"L\n";
			else cout<<"R\n";
		} 
		t=u;
	}
	
	return ;
}

int main() {
	std::ios::sync_with_stdio(false);
	int T;
	cin>>T;
	while(T--) {
		slv();
	}
	
	return 0;
} 
举报

相关推荐

0 条评论