现在有n位工程师和6项工作(编号为0至5),现在给出每个人能够胜任的工作序号表(用一个字符串表示,比如:045,表示某位工程师能够胜任0号,4号,5号工作)。现在需要进行工作安排,每位工程师只能被安排到自己能够胜任的工作当中去,两位工程师不能安排到同一项工作当中去。如果两种工作安排中有一个人被安排在的工作序号不一样就被视为不同的工作安排,现在需要计算出有多少种不同工作安排计划。
输入描述:
输入数据有n+1行:
第一行为工程师人数n(1 ≤ n ≤ 6)
接下来的n行,每行一个字符串表示第i(1 ≤ i ≤ n)个人能够胜任的工作(字符串不一定等长的)
输出描述:
输出一个整数,表示有多少种不同的工作安排方案
输入例子1:
6
012345
012345
012345
012345
012345
012345
输出例子1:
720
package test02;
import java.util.*;
//使用图的深度优先搜索
public class Main05 {
static int count=0;//记录工作安排的方案总数
static int n;
static int[]mid;//定义一个一维数组mid,mid[i]=1表示工作已分配出去
static int[][]num;//定义一个二维数组num横坐标表示工程师编号,纵坐标表示工作编号
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
while (in.hasNext()) {
n = in.nextInt();
num=new int[n][6];
mid=new int[6];
for (int i = 0; i < n; i++) {
String str=in.next();
char[] array = str.toCharArray();
for(int j=0;j<array.length;j++){
num[i][array[j]-'0']=1;//num[i][j]=1表示第i个工程师可以胜任第j项工作
}
}
dfs(0);
System.out.println(count);
}
}
//i表示工程师编号
public static void dfs(int i){
if(i==n){//当i==n表示所有人都已经遍历完,此时count+1;
count++;
return;
}
for(int j=0;j<6;j++){
//表示第j向工作没有分配出去,并且第i个工程师可以胜任第j项工作
if(mid[j]==0&&num[i][j]==1){
mid[j]=1;
dfs(i+1);
mid[j]=0;
}
}
}
}
注意图的深度优先和广度优先的区别
注意图的深度优先和广度优先的应用场景
1、深度优先适合求是否有解(比如这一题,求解的个数),不适合求最优解,比如最短路径
2、广度优先适合求最优解,比如要求迷宫的左上角到右下角的最短路线。广度优先需要对列,缺点也很明显,当树的层次很深并且节点很多的时候队列消耗的空间比较大。