【J1】【搜索】网格寻路

题目描述

给你一个 n * m 的网格，其中每个单元格不是 0（空）就是 1（障碍物）。每一步，您都可以在空白单元格中上、下、左、右移动。

如果您最多可以消除 k 个障碍物，请找出从左上角 (1, 1) 到右下角 (n, m) 的最短路径，并返回通过该路径所需的步数。如果找不到这样的路径，则返回 -1 。

输入

第一行，两个整数n，m

接下来的n行，每行输入一个整数0或1

第n+2行一个整数k，表示最多可以消除的障碍物数目

输出

输出从起点到终点的最短路程，如果不存在从起点到终点的路，则输出-1。

样例输入

5 3
0 0 0
1 1 0
0 0 0
0 1 1
0 0 0
1

样例输出

6

提示

起点和终点坐标的值为0

1≤n,m≤40

k≤n*m

来源

#include<bits/stdc++.h>
using namespace std;
bool vis[45][45][82],a[60][60],flag;
int n,m,k,dx[4]={-1,0,1,0},dy[4]={0,1,0,-1};
struct node{int x,y,s,k;};
queue<node>q;
void dfs(int x,int y){
	queue<node>q;
	node sp={1,1,0,k};
	q.push(sp);
	vis[x][y][k]=1;
	while(!q.empty()){
		node q2=q.front();
		q.pop();
		if(q2.x==n&&q2.y==m){cout<<q2.s;flag=1;return;}
		for(int i=0;i<4;i++){
			int xx=dx[i]+q2.x,yy=dy[i]+q2.y;
			if(xx>=1&&xx<=n&&yy>=1&&yy<=m&&a[xx][yy]==0&&vis[xx][yy][q2.k]==0){
				node q3={xx,yy,q2.s+1,q2.k};
				q.push(q3);
				vis[xx][yy][q2.k]=1;
			}
			else if(xx>=1&&xx<=n&&yy>=1&&yy<=m&&a[xx][yy]==1&&q2.k>0&&vis[xx][yy][q2.k-1]==0){
				node q3={xx,yy,q2.s+1,q2.k-1};
				q.push(q3);
				vis[xx][yy][q2.k-1]=1;
			}
		}
	}
}
int main(){
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=m;j++){
			cin>>a[i][j];
		}
	}cin>>k;
	k=min(n+m-3,k);dfs(1,1);
	if(!flag){cout<<-1;}
	return 0;}