蓝桥杯算法题解 历届试题 分考场

题目描述

问题描述
　　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

题解：

直接看代码的注释。
**注意： **必须要有剪枝，否则只有40分。

代码：

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <climits>
#include <cstring>
#include <string>
#include <algorithm>
#include <vector>
#include <deque>
#include <list>
#include <utility>
#include <set>
#include <map>
#include <stack>
#include <queue>
#include <bitset>
#include <iterator>
using namespace std;

typedef long long ll;
const int inf = 0x3f3f3f3f;
const ll  INF = 0x3f3f3f3f3f3f3f3f;
const double PI = acos(-1.0);
const double E = exp(1.0);
const int MOD = 1e9+7;
const int MAX = 1e2+5;

int n,m;
int a,b;
int stuMap[MAX][MAX] = {0};// 学生之间的映射关系（stuMap[i][j]=1or0）
vector <int> room[MAX];// room[i].size：表示考场i的已有学生人数，room[i][j]：表示考场i的第j个学生
int minRoomNum = inf;// 最少的考场数量

void func(int id,int roomNum)// id：正在安排的学生编号，roomNum：已经分配的考场数量
{
    //printf("学生编号: %d，考场已分配的数量: %d\n",id,roomNum);
    if(roomNum >= minRoomNum)
    {
        // 剪枝：如果已经分配的考场数量大于了当前可以分配的最少考场数量，说明没有必要继续进行递归了
        return;
    }
    if(id == n+1)
    {
        // 当已经分配了n个学生了，就记录最少的考场数量
        minRoomNum = min(minRoomNum,roomNum);
        return;
    }
    // 此时需要为第id个学生分配教室，要满足条件：不跟分配的教室的任何一个人认识
    // 如果满足，直接分配到这个考场room[i].push_back(id);
    // 如果不满足，说明需要重新开一个考场，room[roomNum+1].push_back(id);
    for(int i = 1; i <= roomNum; i++)
    {
        bool flag = true;
        for(int j = 0; j < room[i].size(); j++)
        {
            if(stuMap[id][room[i][j]] == 1)
            {
                flag = false;
                break;
            }
        }
        if(flag)// 对于学生id，不认识考场i的任何一个人，可以直接分配到这个考场
        {
            room[i].push_back(id);
            func(id+1,roomNum);
            // 回溯：将这个学生拿出这个考场，重新分配
            vector<int>::iterator tmp = room[i].end(); tmp--;
            room[i].erase(tmp);
        }
    }
    // 如果前面没有回溯进去，说明学生id和roomNum个考场都有认识的人，要重新开个考场
    room[roomNum+1].push_back(id);
    func(id+1,roomNum+1);
    vector<int>::iterator tmp = room[roomNum+1].end(); tmp--;
    room[roomNum+1].erase(tmp);
}

int main()
{
    /*
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    */
    cin >> n >> m;
    for(int i = 1; i <= m; i++)
    {
        cin >> a >> b;
        stuMap[a][b] = stuMap[b][a] = 1;
    }
    func(1,0);// 从第一个学生开始安排，此时没有分配教室
    cout << minRoomNum << endl;

    return 0;
}