【DFS】蓝桥杯试题 分考场

题目描述

OJ地址 http://lx.lanqiao.cn/problem.page?gpid=T457
　　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

思路：

在做这个题之前，做了一个并查集的题，刚开始读这个题的时候，以为还是一个并查集的，很像。后来仔细想了想不是那回事。还是用dfs做吧。思路如下：

检查第x个学生，看看能不能放入已分配的教室（检查x与前边分配好的学生的关系）

循环判断前n个教室

​ 如果 能安排进当前教室，就分配进去

​ dfs下一个学生

​ 回溯

此外，判断完前n个教室，还可以考虑再新开一个教室（这并不一定会使总的教室数增加，相反，可能会使总教室数减少）。

综上，需要考虑的问题有两个

1. 记录每个教室内已经安排的哪些学生。
2. 判断第x个学生能否安排在第y个教室

这里，存储教室与教室内学生，我使用的是map<int,vector > 。我认为这样比较好理解，first就是教室的编号，后边的vector就是存放的每个已安排的学生。 获取教室数量也很方便。如果使用二维数组的话，还得有其他的变量来判断教室的数量，稍嫌麻烦。

代码

#include <bits/stdc++.h>

using namespace std;
int  n,m;
bool link[105][105];   // 两个学生认不认识 
map<int,vector<int> > room; 
int ans = 9999999;

bool judge(int x, int num){    // 第x个学生能不能放在编号为  num的教室
	 
	 vector<int> v = room[num];  //取出学生列表
	 for(vector<int>::iterator it = v.begin(); it != v.end(); it++){
	 	if(link[x][*it] || link[*it][x] || *it == x ){
	 		return false;
	 	}
	 } 
	 return true;
}


void dfs(int x){    //当前要安排第x个学生

	if(room.size() >= ans){
		return ;
	}
	
	if( x >= n+1){
		if(room.size() < ans){
			ans = room.size();
		}
		return;
	}
	
	
	for(int i = 0; i < room.size(); i++){   // 检查以前安排过的教室 能不能放
		
		if(judge(x,i)){  // 可以放 
			room[i].push_back(x);  // 放进这个教室
			dfs( x+1 ); 
			room[i].pop_back();  // 拿出教室
		}
	} 
	
	// 再开一个教室
	int ii = room.size();
	room[ii].push_back(x); 
	dfs(x+1); 
	room[ii].pop_back();  // 拿出教室
	// 回收这个教室  这里容易忘记这个操作，要在map中删除这条记录。 
	room.erase(ii); 
	
	return;
	
}

int main(){
	
	
	cin >> n >> m;
	

	for(int i = 0; i < m; i++){
		int a,b;
		cin >> a >> b;
		link[a][b] = link[b][a] = true;
	}
	dfs(1);
	cout << ans << endl;
	

	return 0;
}