PAT L2-010 排座位(25 分)

并查集的题目,如果两个人是敌对关系就不需要merge

最终按照题目的要求输出就可以

题目链接

#include <iostream>
#include <cstring>
#define MAXN 1005
using namespace std;
int par[MAXN];
int get_root(int a)
{
	if(par[a]!=a)
	{
		par[a] = get_root(par[a]);
	}
	return par[a];
}
void merge(int a,int b)
{
	int p1 = get_root(a);
	int p2 = get_root(b);
	if(p1==p2)
	{
		return ;
	}
	else
	{
		par[p2] = p1;
		//这里还需要进行其他操作? 
	} 
}
int guanxi[105][105]; 
int main(void)
{
	//初始化区域 
	memset(guanxi,0,sizeof(guanxi));
	for(int i=0;i<MAXN;i++)//初始化啊 
	{
		par[i] = i;	
	} 
	
	int n,m,k;
	scanf("%d%d%d",&n,&m,&k);
	for(int i=0;i<m;i++)
	{
		int a,b,c;
		scanf("%d%d%d",&a,&b,&c);
		if(c!=-1)
			merge(a,b);//把a,b组合到一起
		guanxi[a][b] = guanxi[b][a] = c;	
	} 
	for(int i=0;i<k;i++)
	{
		int x,y;
		scanf("%d%d",&x,&y);	
		int t1 = get_root(x);
		int t2 = get_root(y);
		if(t1==t2&&guanxi[x][y]==-1)//有敌对,但是也有共同的朋友 
		{
			printf("OK but...\n"); 
		}
		else if(t1!=t2&&guanxi[x][y]==-1)//只有敌对
		{
			printf("No way\n"); 
		} 
		else if(guanxi[x][y]==1)
		{
			printf("No problem\n");
		}
		else
		{
			printf("OK\n");
		}
	}
	
}

 

展开阅读全文

没有更多推荐了,返回首页