布置宴席最微妙的事情，就是给前来参宴的各位宾客安排座位。无论如何，总不能把两个死对头排到同一张宴会桌旁！这个艰巨任务现在就交给你，对任何一对客人，请编写程序告诉主人他们是否能被安排同席。
输入格式：
输入第一行给出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表示为何我第一反应是用floyd……
#include <cstdio> #include <cstring> #include <iostream> #include <algorithm> using namespace std; int N, M, K; int u, v, w; //c中，1表示直接朋友关系，0表示没有直接关系，6表示是敌人但是也有间接朋友关系 int c[105][105]; int main() { memset(c, 0, sizeof(c)); for (int i = 1; i <= N; i++) { c[i][i] = 1; } cin >> N >> M >> K; for (int i = 1; i <= M; i++) { scanf("%d%d%d", &u, &v, &w); c[u][v] = w; c[v][u] = w; } for (int k = 1; k <= N; k++) { for (int i = 1; i <= N; i++) { for (int j = 1; j <= N; j++) { if (i != j) { //朋友的朋友是朋友，朋友是递归定义 if ((c[i][k] == 1 || c[i][k] == 6) && (c[k][j] == 1 || c[k][j] == 6)) { if (c[i][j] == -1) { c[i][j] = c[j][i] = 6; } } } } } } while (K--) { scanf("%d%d", &u, &v); if (c[u][v] == 0) { puts("OK"); } else if (c[u][v] == 1) { puts("No problem"); } else if (c[u][v] == -1) { puts("No way"); } else if (c[u][v] == 6) { puts("OK but..."); } } return 0; }
并查集也行
#include <cstdio> #include <cstring> #include <iostream> #include <algorithm> using namespace std; int N, M, K; int u, v, w; int c[105][105]; int f[105]; int getf(int x) { return f[x] == x ? x : f[x] = getf(f[x]); } void Merge(int x, int y) { int u = getf(x), v = getf(y); if (u != v) { if (u < v) { f[v] = u; } else { f[u] = v; } } } int main() { memset(c, 0, sizeof(c)); for (int i = 0; i <= 103; i++) { f[i] = i; } cin >> N >> M >> K; for (int i = 1; i <= M; i++) { scanf("%d%d%d", &u, &v, &w); c[u][v] = w; c[v][u] = w; if (c[u][v] == 1) { Merge(u, v); } } while (K--) { scanf("%d%d", &u, &v); if (c[u][v] == 1) { puts("No problem"); } else if (c[u][v] == 0) { puts("OK"); } else if (c[u][v] == -1) { if (getf(u) == getf(v)) { puts("OK but..."); } else { puts("No way"); } } } return 0; }
