L2-010 排座位(C++, 分数 35)

布置宴席最微妙的事情，就是给前来参宴的各位宾客安排座位。无论如何，总不能把两个死对头排到同一张宴会桌旁！这个艰巨任务现在就交给你，对任何一对客人，请编写程序告诉主人他们是否能被安排同席。

输入格式：

输入第一行给出3个正整数：N（≤100），即前来参宴的宾客总人数，则这些人从1到N编号；M为已知两两宾客之间的关系数；K为查询的条数。随后M行，每行给出一对宾客之间的关系，格式为：宾客1 宾客2 关系，其中关系为1表示是朋友，-1表示是死对头。注意两个人不可能既是朋友又是敌人。最后K行，每行给出一对需要查询的宾客编号。

这里假设朋友的朋友也是朋友。但敌人的敌人并不一定就是朋友，朋友的敌人也不一定是敌人。只有单纯直接的敌对关系才是绝对不能同席的。

输出格式：

对每个查询输出一行结果：如果两位宾客之间是朋友，且没有敌对关系，则输出No problem；如果他们之间并不是朋友，但也不敌对，则输出OK；如果他们之间有敌对，然而也有共同的朋友，则输出OK but...；如果他们之间只有敌对关系，则输出No way。

输入样例：

7 8 4
5 6 1
2 7 -1
1 3 1
3 4 1
6 7 -1
1 2 1
1 4 1
2 3 -1
3 4
5 7
2 3
7 2

输出样例：

No problem
OK
OK but...
No way

思路：

这个问题可以使用并查集来解决。并查集是一种数据结构，用于处理集合的合并和查找操作。在这个问题中，我们可以将每个宾客看作一个节点，朋友关系可以看作是连接两个节点的边，而敌对关系则表示两个节点之间没有直接的边连接。

首先，我们初始化每个宾客为一个单独的集合，即每个宾客的根节点是其自身。然后，根据已知的朋友关系，我们可以不断地合并集合，即将某个宾客所在的集合根节点更新为另一个宾客所在集合的根节点，这样就可以实现朋友关系的合并操作。

同时，我们还需要记录宾客之间的敌对关系。这里使用一个二维数组来记录，数组中的元素 d[i][j] 表示第 i 个宾客和第 j 个宾客之间是否是敌对关系。

接下来，对于每个查询，我们通过并查集找到两个宾客所在的集合的根节点，然后判断它们之间的关系：如果它们在同一个集合且没有敌对关系，就输出 "No problem"；如果它们不在同一个集合但也没有敌对关系，就输出 "OK"；如果它们在同一个集合但有敌对关系，就输出 "OK but..."；如果它们之间只有敌对关系，就输出 "No way"。

代码：

#include <iostream>

using namespace std;

const int N = 10010;

int n, m, k;
int pfriend[N]; // 并查集数组，记录每个人的朋友关系
int d[105][105]; // 二维数组，记录每对人之间的敌对关系

// 并查集的查找函数
int find_friend(int u) {
    if (pfriend[u] != u) pfriend[u] = find_friend(pfriend[u]);
    return pfriend[u];
}

int main() {
    // 初始化并查集
    for (int i = 0; i < N; i++) pfriend[i] = i;
    
    // 输入参宴宾客总人数、已知关系数、查询条数
    cin >> n >> m >> k;
    
    // 处理已知的宾客关系
    for (int i = 0; i < m; i++) {
        int a, b, c;
        cin >> a >> b >> c;
        if (c == 1) pfriend[find_friend(a)] = find_friend(b); // 如果是朋友，更新并查集
        else {
            d[a][b] = d[b][a] = 1; // 如果是敌对关系，记录在二维数组中
        }
    }
    
    // 处理查询
    for (int i = 0; i < k; i++) {
        int af, bf, a1, b1;
        cin >> af >> bf;
        a1 = af, b1 = bf;
        af = find_friend(af), bf = find_friend(bf); // 查找宾客所属的集合
        if (af == bf && !d[a1][b1]) puts("No problem"); // 如果在同一集合且没有敌对关系，输出 No problem
        else if (af != bf && !d[a1][b1]) puts("OK"); // 如果不在同一集合但也没有敌对关系，输出 OK
        else if (af == bf && d[a1][b1]) puts("OK but..."); // 如果在同一集合但有敌对关系，输出 OK but...
        else puts("No way"); // 其他情况输出 No way
    }
}