问题描述
n个人参加某项特殊考试。
为了公平,要求任何两个认识的人不能分在同一个考场。
求是少需要分几个考场才能满足条件。
输入格式
第一行,一个整数n(1<n<100),表示参加考试的人数。
第二行,一个整数m,表示接下来有m行数据
以下m行每行的格式为:两个整数a,b,用空格分开 (1<=a,b<=n) 表示第a个人与第b个人认识。
输出格式
一行一个整数,表示最少分几个考场。
样例输入1
5
8
1 2
1 3
1 4
2 3
2 4
2 5
3 4
4 5
样例输出1
4
样例输入2
5
10
1 2
1 3
1 4
1 5
2 3
2 4
2 5
3 4
3 5
4 5
样例输出2
5
解题思路:
这题的数据量不是很大,可以用深搜回溯解决;记录两个人是否相识可以用一个n*n的二维数组,如果第i个人和第j个人相识,则把arr[i][j]和arr[j][i]标记为1;类似于用邻接矩阵表示无向图。
这个深搜和经典的深搜不太一样,从深搜树上看每一个节点的选择都不同;第一个人只有一种选择即新建立以教室,第二个人有两个选择新建一个教室或者加入第一个人的教室(假如没有冲突)。假设当前的教室个数为m,则第n个人在没有冲突的情况下就有m+1个选择。
为了提高效率,可以在深搜的过程中记录教室个数曾经的最小值,当统计的教室个数超过了记录的最小值,则回溯(剪枝)。因为既然当前的教室个数已经超过了记录的最小值,则表示按照这个方式接着深搜下去也不会得到更小的答案。
解题代码:
import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;
public class 分考场2 {
static int n; //人数
static int m; //表示数据的行数
static int[][] arr; //用于记录两个人是否相识
static List<ArrayList<Integer>> room = new ArrayList<ArrayList<Integer>>();; //存储教室
static int min = Integer.MAX_VALUE; //用于统计教室的最小值
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
n = scanner.nextInt();
m = scanner.nextInt();
arr = new int[n+1][n+1]; //从1开始使用
for(int i=0; i<m; i++){
int a = scanner.nextInt();
int b = scanner.nextInt();
arr[a][b] = 1; //因为相识是相互的,所以要标记两个位置
arr[b][a] = 1;
}
dfs(1);
System.out.println(min);
}
//深搜尝试添加教室
private static void dfs(int index) {
if(index>n){ //递归出口
min = room.size();//因为有剪枝,所以这里就不用判断min是否比room.size()大了
return ;
}
if(index==1){
ArrayList<Integer> temp = new ArrayList<Integer>();
temp.add(1);
room.add(temp);
dfs(index+1);
}else{
//尝试能否添加其中一个教室
int size = room.size();
for(int i=0; i<size; i++){
ArrayList<Integer> temp2 = room.get(i);
if(check(index,temp2)){ //检查能否安排在这个教室
temp2.add(index);
dfs(index+1);
temp2.remove(temp2.indexOf(index)); //回溯
}
}
//如果所有的教室都不能添加,则新建一个教室
ArrayList<Integer> newroom = new ArrayList<Integer>();
newroom.add(index);
room.add(newroom);
if(room.size()>=min){ //剪枝 , 剪枝不能放在函数的开头,否则得到的答案不正确
room.remove(room.indexOf(newroom)); //回溯
return ;
}
dfs(index+1);
room.remove(room.indexOf(newroom)); //回溯
}
}
//判断第index个人是否能添加进教室t
private static boolean check(int index, ArrayList<Integer> t) {
for(Integer i:t){
if(arr[i][index]==1 || arr[index][i]==1){
return false;
}
}
return true;
}
}