蓝桥杯 历届试题 分考场

问题描述

  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;
	}
}

测试结果：