L2-010 排座位

L2-010 排座位 (25 分)

题目链接

题意

思路

1. 利用并查集判断是是否为朋友的情况，因为朋友的朋友也是朋友，利用二维数组判断是否为敌对关系，然后再根据不同的情况输出答案

坑点

1. 两人的关系判断不一定是左右固定需要同时判断两种二维数组都是敌对

代码

#include<bits/stdc++.h>
using namespace std;
int fa[1100];
int a1[110][110]={0};//判断敌对关系
int find(int x)//利用并查集查找
{
    if(x==fa[x])
    {
        return x;
    }
    else{
        fa[x]=find(fa[x]);
        return fa[x];
    }
}
void unio(int i,int j)//合并
{
    int x1=find(i);
    int x2=find(j);
    fa[x1]=x2;
}
int main()
{
    int n,m,k;
    cin>>n>>m>>k;
    for(int i=1;i<=n;i++)//初始化
    {
        fa[i]=i;
    }
    for(int i=0;i<m;i++)
    {
        int a,b,c;
        cin>>a>>b>>c;
        if(c==1)//如果为朋友合并
        {
            unio(a,b);
        }
        else{//如果为敌人要特判
            a1[a][b]=1;//两人是敌对关系，顺序关系
            a1[b][a]=1;
        }
    }
    while(k--)
    {
        int a,b;
        cin>>a>>b;
        if(a1[a][b]==0&&a1[b][a]==0&&find(a)==find(b))//如果两位宾客之间是朋友，且没有敌对关系
        {
            cout<<"No problem"<<endl;
        }
        else if(a1[a][b]==0&&a1[b][a]==0&&find(a)!=find(b))//如果他们之间并不是朋友，但也不敌对
        {
            cout<<"OK"<<endl;
        }
        else if(find(a)==find(b)&&a1[a][b]==1)//如果他们之间有敌对，然而也有共同的朋友
        {
            cout<<"OK but..."<<endl;
        }
        else if(a1[a][b]==1&&a1[b][a]==1)//如果他们之间只有敌对关系
        {
            cout<<"No way"<<endl;
        }
    }
    return 0;
}