萌新一枚,主要是想记录自己的刷题生涯,有不对的地方望大神斧正。
题目:
布置宴席最微妙的事情,就是给前来参宴的各位宾客安排座位。无论如何,总不能把两个死对头排到同一张宴会桌旁!这个艰巨任务现在就交给你,对任何一对客人,请编写程序告诉主人他们是否能被安排同席。
输入格式:
输入第一行给出3个正整数:N
(≤100),即前来参宴的宾客总人数,则这些人从1到N
编号;M
为已知两两宾客之间的关系数;K
为查询的条数。随后M
行,每行给出一对宾客之间的关系,格式为:宾客1 宾客2 关系
,其中关系
为1表示是朋友,-1表示是死对头。注意两个人不可能既是朋友又是敌人。最后K
行,每行给出一对需要查询的宾客编号。
这里假设朋友的朋友也是朋友。但敌人的敌人并不一定就是朋友,朋友的敌人也不一定是敌人。只有单纯直接的敌对关系才是绝对不能同席的。
输出格式:
对每个查询输出一行结果:如果两位宾客之间是朋友,且没有敌对关系,则输出No problem
;如果他们之间并不是朋友,但也不敌对,则输出OK
;如果他们之间有敌对,然而也有共同的朋友,则输出OK but...
;如果他们之间只有敌对关系,则输出No way
。
时间限制
200 ms
内存限制
64 MB
这道题初看很懵,原本一看到题目就想到并查集(不懂的小伙伴请自行在CSDN里搜,因为这只是我记录刷题的文章,并不讲高深的东西,只是提供思路,再说算法早就在CSDN不值钱了,哪里不懂搜哪里),但他却有敌对关系。后来在看到输出格式时,才发现这就是并查集的运用,只不过多了一个判断的敌对关系的条件罢了。怎么理解呢?如果我们判断条件分成朋友关系和敌对关系,按照输出要求,我只需要判断两者的是与否就行了。听起来很废话,但这样分,朋友关系就变成了标准的并查集形式,直接套公式即可。闲话少说,直接上代码:
#include<iostream>
using namespace std;
int node[105];
bool flag[105][105];
int find(int x)
{
if (node[x] != x) node[x]=find(node[x]);
return node[x];
}
void build(int x, int y)
{
x = find(x); y = find(y);
node[x] = y;
}
void init(int n)
{
for (int i = 1; i <= n; i++) node[i] = i;
}
int main()
{
int n, m, k,x,y,b;
cin >> n >> m >> k;
init(n);
for (int i = 0; i < m; i++)
{
cin >> x >> y >> b;
if (b == 1) build(x, y); //建立朋友关系
else flag[x][y] = flag[y][x] = true; //若存在敌对关系,用二维数组记录,表明x对y,y对x存在敌对关系。
}
for (int i = 0; i < k; i++)
{
cin >> x >> y;
bool flag1 = find(x) == find(y); //判断是否具有同一朋友
if (flag1 && flag[x][y] == false) cout << "No problem" << endl;
else if (!flag1&&flag[x][y] == false) cout << "OK" << endl;
else if (flag1&&flag[x][y] == true) cout << "OK but..." << endl;
else cout << "No way" << endl;
}
}