现在有n位工程师和6项工作(编号为0至5),现在给出每个人能够胜任的工作序号表(用一个字符串表示,比如:045,表示某位工程师能够胜任0号,4号,5号工作)。现在需要进行工作安排,每位工程师只能被安排到自己能够胜任的工作当中去,两位工程师不能安排到同一项工作当中去。如果两种工作安排中有一个人被安排在的工作序号不一样就被视为不同的工作安排,现在需要计算出有多少种不同工作安排计划。
输入描述:
输入数据有n+1行: 第一行为工程师人数n(1 ≤ n ≤ 6) 很重要,工作人数没有超过工作总数,可以使用dfs,终止条件为i==工程师人数 接下来的n行,每行一个字符串表示第i(1 ≤ i ≤ n)个人能够胜任的工作(字符串不一定等长的)
输出描述:
输出一个整数,表示有多少种不同的工作安排方案
输入例子:
6 012345 012345 012345 012345 012345 012345
输出例子:
720----------------------------------------------------------------------------------------------------------------------------------------------------
思路:
首先现根据输入的字符串创建二位数组,横坐标为人,纵坐标为工作,可以做为1,不能做为0,进行深搜(dfs),i代表遍历到第i个人,当i=sum时,所以人均以遍历完成,工作分配完成。s从工作0到工作5进行遍历,创建数组mid存储已经分配出去的工作,1为已分配,0为未分配,当工作未分配且第i个人可以完成,mid值给1,递归调用dfs给i+1个人分配工作,当所有i+1均以搜索完成,dfs结束,mid[s]给0,接着沿着s进行遍历。
代码:
import java.util.Scanner;
public class Main {
static int count = 0;
static int sum;
static int[][] vector;
static int mid[];
public static void main(String args[]){
Scanner sc = new Scanner(System.in);
sum = sc.nextInt();
String ss = sc.nextLine();
vector = new int[sum][6];
mid = new int[6];
for(int i=0;i<sum;i++){
String s = sc.nextLine();
char[] array = s.toCharArray();
for(int k=0;k<array.length;k++)
vector[i][array[k]-'0']=1;
}
dfs(0);
System.out.println(count);
}
private static void dfs(int i) {
if(i==sum){
count++;
return;
}
//s代表工作,遍历每个人都擅长的6项工作
for(int s=0;s<6;s++){
//保证分配给工人的工作都是未分配且工人擅长的。
if(mid[s]==0&&vector[i][s]==1){
//mid保存已经分配过的工作
mid[s]=1;
//dfs写这里,是因为在给第i个分配工作后,
//需要给第i+1个分配,第i+1分配后,需要给i+1+1分配
dfs(i+1);
//在递归结束后,将原来分配的工作变成为分配,
mid[s]=0;
}
}
}
}
参考链接: 参考一