L2-010 排座位（并查集）

题面：

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

输入格式：

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

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

输出格式：

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

思路：

要把产生关系（是朋友）的一群人归入一个集合中，从而后续查询时能够找到是否为朋友。对于敌对关系，如果是敌对的，那么在二维数组中记录，使用二维数组能够记录对称关系，从而后续每一组查询的俩人（p1，p2）不用考虑先后直接能在数组中找到。

结果有四种：是朋友->在一个集合中；不是朋友又不是敌人->不在一个集合中并且敌对关系未被标记；之间有敌对也有共同的朋友->在一个集合且被标记；只有敌对关系->被标记。

并查集步骤：1.初始化。2.查询。3.合并。

int r[200];

void init(){
	for(int i = 1;i <= 105;i++)	r[i] = i;
}
int find(int x){
	if(x != r[x]) r[x] = find(r[x]);
	return r[x];
}

void join(int x, int y){
	x = find(x);
	y = find(y);
	if(x != y) r[x] = r[y];
}

敌对关系数组：

int en[200][200];

代码如下：

signed main(){
	   //freopen("C:\\Users\\pengyu\\Desktop\\cin.txt","r",stdin);
	   int n, nn, k;
	   cin >> n >> nn >> k;
	   
	   init();

		while(nn--){
	   		int p1, p2, rela;
	   		cin >> p1 >> p2 >> rela;
	   		
	   		if(rela == 1) 
			   join(p1, p2); //直接或间接朋友关系都进行合并
	   		else 
			   en[p1][p2] = en[p2][p1] = 1; //标记(p1,p2)和(p2,p1)
		}
	   
	   
	  
	   while(k--){
	   	int p1, p2;
	   	cin >> p1 >> p2;
	   	
	   	int x = find(p1), y = find(p2);

	   	if(x == y && en[p1][p2] == 0)
	   		puts("No problem");
	   	else if(x != y && en[p1][p2] == 0)
	   		puts("OK");
	   	else if(x == y && en[p1][p2] == 1)
	   		puts("OK but...");
	   	else if(en[p1][p2] == 1)
	   		puts("No way");
	   }
	   return 0;
}