蓝桥杯——分考场（回溯/dfs）

问题描述

　　n个人参加某项特殊考试。
　　为了公平，要求任何两个认识的人不能分在同一个考场。
　　求是少需要分几个考场才能满足条件。

输入格式

　　第一行，一个整数n(1<n<100)，表示参加考试的人数。
　　第二行，一个整数m，表示接下来有m行数据
　　以下m行每行的格式为：两个整数a，b，用空格分开 (1<=a,b<=n) 表示第a个人与第b个人认识。

输出格式

　　一行一个整数，表示最少分几个考场。

样例输入

5
8
1 2
1 3
1 4
2 3
2 4
2 5
3 4
4 5

样例输出

4

样例输入

5
10
1 2
1 3
1 4
1 5
2 3
2 4
2 5
3 4
3 5
4 5

样例输出

5

dfs版本   （自己敲的就超时，题解的代码就刚好过了）

#include<stdio.h>
#include<string.h>
#define min(a,b) a>b?b:a 
#define N 301
int vis[N][N];//关系表 
int p[N][N];// 房间状态  
int ans=N,n,m;

void dfs(int x,int num){//x 代表当前安排了多少个人 num 代表考场数
    if(num>=ans)
	    return;//剪子 
    if(x==n+1){
	    ans=min(ans,num);
		return;
	}//如果已经安排了n个人，进行判断  
	int j,k;
	for(j=1;j<=num;j++) { //枚举考场
        k=0;
	    while(p[j][k]&&!vis[x][p[j][k]])
		    k++;//找到一个空位 并且与该考场人无关系 
	    if(p[j][k]==0){
	    	p[j][k]=x;
			dfs(x+1,num);
			p[j][k]=0;//满足条件 进行下一考生 
		}
    } //回溯 
	p[j][0]=x;
	dfs(x+1,num+1);// 如果所有房间都不满足条件 增加房间 
	p[j][0]=0;//回溯 
}

int main(){
	int a,b;
	memset(vis,0,sizeof(vis));
	memset(p,0,sizeof(p));
	scanf("%d%d",&n,&m);
	for(int i=1;i<=m;i++){ 
	    scanf("%d%d",&a,&b);
	    vis[s1][s2]=1;
		vis[s2][s1]=1;//建关系 
	}
	dfs(1,1);
	printf("%d\n",ans);
	return 0;
}

回溯版本：

#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
#include<stack>
#include<set>
#include<map>
#include<vector>
#include<cmath>
using namespace std;
int vis[105][105];//联系 
int room[105][105];
int cnt[105];//教室里学生是数量 
int ans=1e9;
int n,m; 
 
void sovle(int id,int num){
	if(num>=ans)
	    return ;
	if(id>n){//全部学生都安排好了 
		ans=min(ans,num);
		return ;
	}
	for(int i=1;i<=num;i++){
		int tmp=cnt[i];
		int res=0;
		for(int j=1;j<=tmp;j++){
			if(vis[id][room[i][j]]==0)
			    res++;
		}
		if(res==tmp){//和这个房间里的都不认识 
		    cnt[i]++;
			room[i][cnt[i]]=id;
			sovle(id+1,num);
			cnt[i]--; 
		}
	} 		
	//得新开一个房间了 
	cnt[num+1]++;
	room[num+1][cnt[num+1]]=id;
	sovle(id+1,num+1);
	cnt[num+1]--;
}
 
int main(){
    scanf("%d%d",&n,&m);
    while(m--){
    	int a,b;
    	scanf("%d%d",&a,&b);
    	vis[a][b]=1;
    	vis[b][a]=1;//建立联系 
	}
	sovle(1,0);
	printf("%d\n",ans);
    return 0;
}