8.dfs--王子救公主(遍歷迷宮,王子和公主訪問過同一位置,即成功救公主)

C.&H.發表於2020-11-05

題目:

分析:

和前面的迷宮例題類似,遇到牆則不能通過

(1)分別尋找王子和公主的初始位置

(2)設定一個三維訪問陣列v[n][m][2]:v[n][m][0]標記王子是否走過、v[n][m][1]標記公主是否走過

(3)分別對王子和公主進行深度優先搜尋dfs(當前x,y,0或1),0表示王子,1表示公主

(4)最後遍歷訪問陣列v,

            如果存在位置i,j,使得v[i][j][0]和v[i][j][1]都為true(即王子和公主都訪問了該位置),則王子和公主肯定相遇(王子能找到公主)

dfs: (1)先判斷當前位置是否合法:是否越界、是否為牆、是否已訪問

    (2)設定當前位置已訪問

    (3)再訪問下一個位置(王子移動2步,公主移動1步)

程式碼:

import java.util.Scanner;

public class Main {
	/*
	 * 思路:
測試資料
4
1 2
1 3
2 4

正常輸出:
3 1 0 0 
	 * */
	static int n;//迷宮的行列
	static int m;
	static char[][] g;//迷宮
	static boolean[][][] v;//訪問陣列
	public static void main(String[] args){
		//1.輸入相關值
		Scanner sc = new Scanner(System.in);
		n = sc.nextInt();
		m = sc.nextInt();
		g = new char[n][];
		v = new boolean[n][m][2];
		//2.初始化迷宮
		for(int i=0;i<n;i++) {
				g[i] = sc.next().toCharArray();
		}
		//3.尋找王子的位置
		int x1=0,y1=0;
		for(int i=0;i<n;i++) {
			for(int j=0;j<m;j++) {
				if(g[i][j]=='w') {
					x1=i;
					y1=j;
					break;
				}
			}
		}
		//4.尋找公主的位置
		int x2=0,y2=0;
		for(int i=0;i<n;i++) {
			for(int j=0;j<m;j++) {
				if(g[i][j]=='g') {
					x2=i;
					y2=j;
					break;
				}
			}
		}
		//5.對王子進行dfs。0表示王子
		dfs(x1,y1,0);
		//6.對公主進行dfs.1表示公主
		dfs(x2,y2,1);
		//7.判斷王子和公主是否訪問過同一位置,即是否相遇
		boolean ans=false;
		for(int i=0;i<n;i++) {
			if(ans) break;
			for(int j=0;j<m;j++) {
				if(v[i][j][0] && v[i][j][1]) {
					ans=true;
					break;
				}
			}
		}
		
		if(ans) {
			System.out.println("YES");
		}else {
			System.out.println("NO");
		}
	}
	/*
	 * x、y:當前開始位置
	 * t:標記是王子還是公主
	 * */
	private static void dfs(int x, int y, int t) {
		//1.判斷位置是否合法
		if(x<0 || x>=g.length || y<0 || y>=g[0].length || g[x][y]=='#' || v[x][y][t]) {
			return;
		}
		//2.訪問陣列
		v[x][y][t]=true;
		//3.訪問下一個陣列
		dfs(x+(2-t),y,t);
		dfs(x-(2-t),y,t);
		dfs(x,y+(2-t),t);
		dfs(x,y-(2-t),t);
	}
	
	
	
}
	

相關文章