pta L2-010 排座位

题目链接：https://pintia.cn/problem-sets/994805046380707840/problems/994805066135879680

一道并查集版子题，还是蛮好做的；

等明天比完赛后总结一下并查集

Talk is cheap. Show me the code.

#include<bits/stdc++.h>
using namespace std;
int n,m,k;
int re[110][110];//relationsh关系 
 //int height[110]; //树的高度 
int s[110];//集合 
void init_set()//初始化 
{
    for(register int i=1;i<=n;i++)
    {
        s[i]=i;//所有集合按顺序初始化 
        //height[i]=0;
    }
    
}
int find_set(int x)//查找函数，路径压缩 
{
    if(x!=s[x])
    s[x]=find_set(s[x]);
    return s[x];
}
void union_set(int x,int y)//合并函数 
{
    x=find_set(x);
    y=find_set(y);
    if(x!=y)
    {
        s[x]=s[y];
    }
}

/*void union_set(int x,int y)//合并函数优化 
{
    x=find_set(x);
    y=find_set(y);
    if(height[x]==height[y])
    {
        height[x]=height[x]+1;
        s[y]=x;
    }
    else
    {
        if(height[x]<height[y])
        {
            s[x]=y;
        }
        else
        s[y]=x;
    }
}*/
int main()
{
    cin>>n>>m>>k;
    init_set(); 
    while(m--)
    {
        int a,b,c;
        cin>>a>>b>>c;
        re[a][b]=re[b][a]=c;
        if(re[a][b]==1||re[b][a]==1)//如果是朋友 
        union_set(a,b);//加到同一个集合 
    }
    for(register int i=1;i<=k;i++)
    {
        int a,b;
        cin>>a>>b;
        if(re[a][b]==1)
        {
            printf("No problem");
        }
        if(re[a][b]==0&&find_set(a)!=find_set(b))
        {
            printf("OK");
        }
        if(re[a][b]==-1&&find_set(a)==find_set(b))
        {
            printf("OK but...");
        }
        if(re[a][b]==-1&&find_set(a)!=find_set(b))
        {
            printf("No way");
        }
        if(i<k)
        printf("\n");
    }
    return 0;
}