当前位置: 代码迷 >> 综合 >> POJ1321 棋盘问题 深度优先搜索JAVA
  详细解决方案

POJ1321 棋盘问题 深度优先搜索JAVA

热度:21   发布时间:2023-12-29 09:12:35.0

Description

在一个给定形状的棋盘(形状可能是不规则的)上面摆放棋子,棋子没有区别。要求摆放时任意的两个棋子不能放在棋盘中的同一行或者同一列,请编程求解对于给定形状和大小的棋盘,摆放k个棋子的所有可行的摆放方案C。

Input

输入含有多组测试数据。 每组数据的第一行是两个正整数,n k,用一个空格隔开,表示了将在一个n*n的矩阵内描述棋盘,以及摆放棋子的数目。 n <= 8 , k <= n 当为-1 -1时表示输入结束。 随后的n行描述了棋盘的形状:每行有n个字符,其中 # 表示棋盘区域, . 表示空白区域(数据保证不出现多余的空白行或者空白列)。 

Output

对于每一组数据,给出一行输出,输出摆放的方案数目C (数据保证C<2^31)。

Sample Input

2 1
#.
.#
4 4
...#
..#.
.#..
#...
-1 -1

Sample Output

2
1
import java.util.*;
public class Main {static int [][]a;static int []col;//标记列是否访问static int count;static int n,k;public static void main(String[] args) {// write your code hereScanner in=new Scanner(System.in);while (in.hasNext()){count=0;n=in.nextInt();k=in.nextInt();if(n==-1 && k==-1)return;a=new int[n+1][n+1];col=new int[n+1];for(int i=0;i<n;i++){String str=in.next();for(int j=0;j<n;j++){char ch=str.charAt(j);if (ch=='.')a[i][j]=0;else if(ch=='#')a[i][j]=1;}}dfs(0,k);System.out.println(count);}}public static void dfs(int index,int num)//index表示行,num表示还剩多少棋子{if(n-index<num) return;//剪枝if(num==0){count++;return;}for(int i=0;i<n;i++){if(a[index][i]==0)continue;if(col[i]==1)continue;col[i]=1;dfs(index+1,num-1);col[i]=0;//注意回溯}dfs(index+1,num);//表示这行不放棋子}
}