【POJ】3984 迷宮問題(入門)
- 思路: 求最短路徑一般就是廣度優先搜索,定義結構體pre[x][y],表示該點的前一個元素,即保存路徑。從左上角(0, 0)位置開始搜索,分別向上下左右搜索,已走過的點標記爲1,未走過且可以走的點壓入隊列,最後當搜索到終點的時候,再把隊列裏的路徑從頭到尾一個個遞歸輸出。
C++ Code:
#include <iostream>
#include <queue>
#define ll long long
using namespace std;
const int N=1e6+5;
int cnt;
int vis[10][10],maps[10][10];
int dir_x[4]={0,0,1,-1};
int dir_y[4]={1,-1,0,0};
struct node{
int x,y;
};
node pre[10][10];
void out(node cur){
if(cur.x==0 && cur.y==0){
cout<<"("<<cur.x<<", "<<cur.y<<")"<<endl;
return;
}
out(pre[cur.x][cur.y]);
cout<<"("<<cur.x<<", "<<cur.y<<")"<<endl;
}
void bfs(){
queue<node> q;
node start;
start.x=0,start.y=0;
q.push(start);
vis[0][0]=1;
while(!q.empty()){
node now = q.front();
q.pop();
if(now.x==4 && now.y==4){
out(now);
return;
}
for(int i=0;i<4;i++){
node next;
next.x=now.x+dir_x[i];
next.y=now.y+dir_y[i];
if(next.x>=0 && next.x<5 && next.y>=0 && next.y<5 && !vis[next.x][next.y] && !maps[next.x][next.y]){
vis[next.x][next.y]=1;
q.push(next);
pre[next.x][next.y]=now;
}
}
}
}
int main(){
for(int i=0;i<5;i++)
for(int j=0;j<5;j++)
cin>>maps[i][j];
bfs();
return 0;
}
Java Code:
import java.util.Scanner;
class node{
int x,y,pre;
public node() {}
public node(int x,int y,int pre) {
this.x=x;
this.y=y;
this.pre=pre;
}
}
public class Main {
static int front=0,rear=1;
static int[][] dir = {{1,0},{-1,0},{0,1},{0,-1}};
static int[][] maps = new int[10][10];
static node[] q = new node[100];
static void print(int index) {
if(q[index].pre!=-1) {
print(q[index].pre);
System.out.println("("+q[index].x+", "+q[index].y+")");
}
}
static void bfs(int x,int y) {
q[front]= new node(x,y,-1);
while(front<rear) {
for(int i=0;i<4;i++) {
for(int j=0;j<4;j++) {
int xx=q[front].x+dir[i][0];
int yy=q[front].y+dir[i][1];
if(xx<0||xx>4||yy<0||yy>4||maps[xx][yy]!=0)
continue;
maps[xx][yy]=1;
q[rear] = new node(xx,yy,front);
if(xx==4&&yy==4) print(rear);
rear++;
}
}
front++;
}
}
public static void main(String[] args) {
Scanner cin = new Scanner(System.in);
for(int i=0;i<5;i++)
for(int j=0;j<5;j++)
maps[i][j]=cin.nextInt();
for(int i=0;i<100;i++) //結構體要全部初始化,不然搜索時會報空指針異常
q[i]=new node();
System.out.println("(0, 0)");
bfs(0,0);
}
}
【HDU】2612 Find a way(兩次BFS)
-
題意: @ 代表目的地,‘.’ 是可以走的路 ,‘#’ 是不可以走的路,有多個目的地,Y和M兩個人要到同一個目的地,問兩個人到同一個目的地的最短路徑和。
-
思路: 兩個人就跑兩遍 BFS,每次 BFS 就記錄到每個目的地所需的最短路徑,最後比較每個目的地得出最小值。注意不能對每個目的地跑兩遍,這樣會TLE。還要考慮的一種特殊情況,就是當某個目的地不能到達時,就不考慮該點。
C++ Code:
#include <iostream>
#include <cstring>
#include <queue>
#define inf 0x3f3f3f3f
using namespace std;
int dir[4][2]={{1,0},{-1,0},{0,1},{0,-1}};
int n,m,x1,y1,x2,y2,ex,ey;
int vis[210][210],ans1[210][210],ans2[210][210];
char maps[210][210];
struct node{
int x,y,step;
};
void bfs(int sx,int sy,char ch){
memset(vis,0,sizeof(vis));
queue<node> q;
while(!q.empty()) q.pop();
node start;
start.x=sx, start.y=sy, start.step=0;
vis[start.x][start.y]=1;
q.push(start);
while(!q.empty()){
node now = q.front();
q.pop();
if(maps[now.x][now.y]=='@'){
if(ch=='Y'){
ans1[now.x][now.y]=now.step;
vis[now.x][now.y]=1;
}
if(ch=='M'){
ans2[now.x][now.y]=now.step;
vis[now.x][now.y]=1;
}
}
for(int i=0;i<4;i++){
int xx=now.x+dir[i][0];
int yy=now.y+dir[i][1];
if(xx>=1 && xx<=n && yy>=1 && yy<=m && !vis[xx][yy] && maps[xx][yy]!='#'){
vis[xx][yy]=1;
node next;
next.x=xx, next.y=yy, next.step=now.step+1;
q.push(next);
}
}
}
}
int main(){
while(cin>>n>>m){
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
cin>>maps[i][j];
if(maps[i][j]=='Y')
x1=i,y1=j;
if(maps[i][j]=='M')
x2=i,y2=j;
}
}
memset(ans1,0,sizeof(ans1));
memset(ans2,0,sizeof(ans2));
bfs(x1,y1,'Y');
bfs(x2,y2,'M');
int ans=inf;
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
if(maps[i][j]=='@' && ans1[i][j] && ans2[i][j])
ans=min(ans,ans1[i][j]+ans2[i][j]);
cout<<ans*11<<endl;
}
return 0;
}