本文共 1447 字,大约阅读时间需要 4 分钟。
为了解决这个问题,我们需要确保在安排宴席座位时,任何一对死对头不会坐在同一张宴会桌旁。我们可以使用并查集(Union-Find)来高效地管理朋友关系,并使用二维数组来记录敌对关系。
#includeusing namespace std;int f[101];int di[101][101];void init() { for (int i = 1; i <= 100; i++) { f[i] = i; }}int find(int x) { if (x == f[x]) { return x; } else { f[x] = find(f[x]); }}void merge(int x, int y) { int a = find(x); int b = find(y); if (a != b) { f[b] = a; }}int main() { int n, m, k; cin >> n >> m >> k; init(); int g1, g2, ship; while (m--) { cin >> g1 >> g2 >> ship; if (ship == 1) { merge(g1, g2); } else { di[g1][g2] = di[g2][g1] = 1; } } while (k--) { cin >> g1 >> g2; int a = find(g1); int b = find(g2); if (di[g1][g2] == 1) { if (a == b) { cout << "OK but..." << endl; } else { cout << "No way" << endl; } } else { if (a == b) { cout << "No problem" << endl; } else { cout << "OK" << endl; } } } return 0;}
init函数初始化并查集,每个宾客最初都是自己的代表。find函数用于查找一个元素的根节点,路径压缩优化了查找时间。merge函数将两个集合合并,用于处理朋友关系。这种方法确保了高效处理大量关系,并准确判断每对宾客是否可以同席。
转载地址:http://njdk.baihongyu.com/