1303: [藍橋杯2016決賽]路徑之謎

dosafdc發表於2020-11-08

題目
在這裡插入圖片描述
在這裡插入圖片描述
思路:感覺就是簡單的dfs,需要一直考慮到回溯,最後還要剪枝,不然會出現超時的問題
AC程式碼:

package 藍橋杯2016;
import java.util.*;
public class 路徑之謎 {
	static Scanner sc=new Scanner (System.in);
	static int X[],Y[];
	static int b[][];//判斷路徑之前是否走過
	static int n;
	static int next[][]= {{1,0},{-1,0},{0,1},{0,-1}};
	public static void main(String[] args) {
		n=sc.nextInt();
		X=new int [n+1];
		Y=new int [n+1];
		b=new int[n+1][n+1];
		for(int i=1;i<=n;i++)Y[i]=sc.nextInt();
		for(int j=1;j<=n;j++)X[j]=sc.nextInt();
		dfs(1,1,1);
	}
	private static void dfs(int x,int y,int s) {//引數為當前位置
		if(x<=0||y<=0||x>n||y>n)return;//越界
		if(b[x][y]!=0)return;//已經走過
		for(int i=1;i<n;i++) if(X[i]<0||Y[i]<0)return;//剪枝,已經出現負,該情況不存在
		X[x]--;
		Y[y]--;
		b[x][y]=s;
		s++;
		if(x==n&&y==n) {
			for(int i=1;i<=n;i++) {
				if(X[i]!=0||Y[i]!=0) {
					X[x]++;
					Y[y]++;
					b[x][y]=0;
					return;
				}
			}
			int a[]=new int [s];
			for(int i=1;i<=n;i++)for(int j=1;j<=n;j++)a[b[i][j]]=(i-1)*n+j-1;
			StringBuilder ss=new StringBuilder();
			for(int i=1;i<s;i++) ss.append(a[i]+" ");
			System.out.println(ss);
		}else for(int i=0;i<4;i++)dfs(x+next[i][0],y+next[i][1],s);
		X[x]++;//回溯
		Y[y]++;
		b[x][y]=0;
	}
}

相關文章