机试题——绩效互评人员分配

题目描述

公司组织绩效互评，为了避免有同学或者同团队的人互相打高分，需要将员工分成两组分别打分。给定一个整数 ( n ) 和一个数组 GoodRelationShips[][]，其中：

• ( n ) 表示人数。
• GoodRelationShips[][] 是一个邻接表，GoodRelationShips[i] 的元素 [a, b, c] 代表员工 ( i ) 和员工 ( a )、( b )、( c ) 是同学或者同团队。

目标是将这 ( n ) 个人分成两组，使得每组不再有同学或者同团队的人。

输入描述

第一行：一个整数 ( n )，表示人数。

接下来 ( n ) 行：每行表示第 ( i ) 个员工与其他员工的关系，格式为 a b c ...，表示员工 ( i ) 与这些员工是同学或同团队。

输出描述

分组方案按照节点编号从小到大排序，输出排序最靠前的一种方案。如果无法分成符合条件的两组，则输出 -1。

用例输入

输入：

4
1 3
0 2
1 3
0 2

输出：

0 2
1 3

解题思路

问题分析

1. 目标：将员工分成两组，使得每组内没有同学或同团队的人。
2. 关键点：
   • 这是一个典型的二分图问题，需要判断给定的图是否是二分图。
   • 如果图是二分图，则可以将员工分成两组；否则，无法满足条件。

算法设计

1. 数据结构：

   • 使用邻接表 g 存储图的结构。
   • 使用数组 cor 存储每个节点的颜色（分组情况），0 表示未染色，1 和 2 表示两种不同的分组。

2. 深度优先搜索（DFS）：

   • 从每个未染色的节点开始，尝试将其染色为 1，并递归地将相邻节点染色为 2。
   • 如果发现相邻节点已经染色且颜色相同，则图不是二分图，返回 -1。

3. 结果输出：

   • 如果图是二分图，将所有节点分为两组，分别存储在 g1 和 g2 中。
   • 按照节点编号从小到大排序，输出两组的节点编号。

代码实现

#include <iostream>
#include <vector>
#include <sstream>
#include <algorithm>
using namespace std;

int n;
vector<int> g[10005];
bool flag = true; // 默认是二分图
vector<int> cor(10005, 0); // 0 表示未染色，1 和 2 表示两种不同的分组

// 当前节点和要染的颜色
void dfs(int root, int c) {
    if (!flag) return;
    cor[root] = c;
    for (int i = 0; i < g[root].size(); i++) {
        int nex = g[root][i];
        if (cor[nex] == c) {
            // 相邻节点颜色一样，有问题
            flag = false;
            return;
        }
        if (cor[nex] == 0) {
            dfs(nex, 3 - c);
        }
    }
}

void print(vector<int> nums) {
    for (int i = 0; i < nums.size(); i++) {
        cout << nums[i];
        if (i != nums.size() - 1) cout << " ";
    }
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    cin >> n;
    cin.ignore();
    for (int i = 0; i < n; i++) {
        string input;
        getline(cin, input);
        istringstream is(input);
        int num;
        while (is >> num) {
            g[num].push_back(i);
            g[i].push_back(num);
        }
    }

    for (int i = 0; i < n; i++) {
        if (cor[i] == 0) dfs(i, 1);
    }

    if (!flag) {
        cout << -1 << endl;
        return 0;
    }

    vector<int> g1, g2;
    for (int i = 0; i < n; i++) {
        if (cor[i] == 1) {
            g1.push_back(i);
        } else {
            g2.push_back(i);
        }
    }

    // 输出序号小的
    if (g1 > g2) {
        swap(g1, g2);
    }

    print(g1);
    cout << "\n";
    print(g2);

    return 0;
}