拯救大兵瑞恩AcWing(bfs最短路+坐标编号化+dp状态转移)

题目

首先要学会横纵坐标转编号
公式：给出(x，y)则它对应一维编号是(x-1)*列+y
给出编号求对应的(x，y) 公式：x=(z-1)/m+1 y=(z-1)%m+1

分析dp部分：这道题对最短路中d数组进行变化，将它的含义变为编号z它目前拥有钥匙数为s，s是一个二进制数eg s=1010即z编号位置拥有钥匙1和钥匙3，那么我们跑一遍bfs先达n*m终点编号的一定是最短路。对于移动的处理，首先遇见墙肯定不能走，遇见门，我们要用手里的钥匙去匹配也就是((s>>x)&1)==1，门的类型为x，如果等于1那么有这把钥匙，如果等于0那么没有这把钥匙，有这把钥匙正常通过，没有这把钥匙不能通过，其他的与bfs最短路问题没有任何差别。

#include <bits/stdc++.h>
using namespace std;
#define INF 0x3f3f3f3f
#define x first
#define y second
typedef pair<int,int> PII;
const int N=105;
const int M=1<<12;
int n,m,ecnt,k,ans,p;
int vis[N][M];//是否拜访过该点 
int d[N][M];//最短路数组记录编号z拥有钥匙k(二进制数)最短路 
int g[N][N];//地图，u--->v u和v都是编号形式
int key[N];//编号z钥匙情况
int dx[4]={-1,0,1,0},dy[4]={0,1,0,-1}; 
queue<PII> q;//存放编号和钥匙情况，用于bfs遍历队列
 
int get(int x,int y){
	 return (x-1)*m+y; 
} 

int bfs(){//bfs最短路 
	int t=get(1,1);//(1,1)为起点
	
	q.push({t,key[t]});
	vis[t][key[t]]=1;
	memset(d,INF,sizeof d);
	d[t][key[t]]=0;
	while(!q.empty()){
		PII t=q.front();
		q.pop();
	    int u=t.first,s=t.second; //起点u，有s钥匙
		
		for(int i=0;i<4;i++){
			int x=(u-1)/m+1+dx[i],y=(u-1)%m+1+dy[i]; //下一步坐标 
		    
		    int ns=s,v=get(x,y);//新的编号v，新的钥匙ns
			
			if(!x||!y||x>n||y>m||!g[u][v])continue;//边界+墙 
			if(g[u][v]!=-1){//门 
				if(!(ns>>g[u][v]&1))continue;//没有钥匙 
			} 
			//可以通过 空地/有钥匙情况
			ns|=key[v];//捡起钥匙
			if(!vis[v][ns]){
				q.push({v,ns});//新加入队列 
				vis[v][ns]=1;
				d[v][ns]=d[u][s]+1;//转移 
			} 
		    if(v==n*m)return d[v][ns];//先到的最短 
		} 
	} 
	
	return -1;//到达不了 	 
} 




int main() {
    
    
    cin>>n>>m>>p; 
    cin>>k;
	int x1,y1,x2,y2,z,z1,z2;
	memset(g,-1,sizeof g);//-1为空地
	while(k--){
		cin>>x1>>y1>>x2>>y2>>z;
		z1=get(x1,y1),z2=get(x2,y2);
		g[z1][z2]=g[z2][z1]=z;//双向边 
	} 
	cin>>k;
	while(k--){
		cin>>x1>>x2>>z;
		key[get(x1,x2)]|=1<<z; //一个地方可能多把钥匙所以是|=不是= 
	} 
	cout<<bfs()<<endl; 












	return 0;
}