pta L2-010 排座位

最新推荐文章于 2023-04-10 18:06:28 发布

原创最新推荐文章于 2023-04-10 18:06:28 发布 · 343 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#图论 #dfs

Pta 专栏收录该内容

9 篇文章

订阅专栏

本文探讨了在宴会上合理安排宾客座位的问题，通过构建图算法解决宾客间朋友与敌对关系，确保宴会和谐。使用邻接矩阵和深度优先搜索（DFS）策略，判断宾客能否同席，涵盖朋友关系传递、敌对关系处理等复杂场景。

L2-010 排座位

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

输入格式：

输入第一行给出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

思路：

该题目的难点在于理解题意，弄懂以上几种情况是如何出现的该题目基本就解决了。利用邻接矩阵，分别按照不同的条件进行dfs或者bfs即可。

AC代码：

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e2 + 5;
int node[maxn][maxn]={0};
bool iswhat[10];
bool vis[maxn];
int v,n,m;
void find(int u,int val){   //val的值为1的时候是查找朋友，-1是查找敌人
	vis[u] = true;
	for(int i=1;i<=v;i++){
		if(node[u][i]==val&&!vis[i]){
			find(i,val);
		}
	}
	return ;
}
//0没有关系 1两人是朋友 2两人是敌人  3没关系  
void judge(int x,int y){
	if(node[x][y]==-1){
		iswhat[2] = true;
		find(x,1);
		if(vis[y]){
			iswhat[1] = true;
			cout<<"OK but..."<<endl;
		}else{
			cout<<"No way"<<endl;
		}
	}else if(node[x][y]==1){
		iswhat[1] = true;
		cout<<"No problem"<<endl;
	}else{
		iswhat[0] = true;
		iswhat[3] = true;
		find(x,1);
		if(vis[y]){
			iswhat[1] = true;
			iswhat[3] = false;
		}
		if(iswhat[0]&&iswhat[3]){
			cout<<"OK"<<endl;
		}else if(iswhat[1]){
			cout<<"No problem"<<endl;
		}
	}
	
}
int main(void){
	cin>>v>>n>>m;
	for(int i=0;i<n;i++){
		int x,y,r;
		cin>>x>>y>>r;
		node[x][y] = r;
		node[y][x] = r;
	}
	fill(iswhat,iswhat+10,false);
	fill(vis,vis+maxn,false);
	for(int i=0;i<m;i++){
		int x,y;
		cin>>x>>y;
		judge(x,y);
		fill(iswhat,iswhat+10,false);
		fill(vis,vis+maxn,false); 
	}
	return 0;
}