洛谷P2256 一中校运会之百米跑

洛谷P2256 一中校运会之百米跑

题目背景

在一大堆秀恩爱的**之中，来不及秀恩爱的苏大学神踏着坚定（？）的步伐走向了100米跑的起点。这时苏大学神发现，百米赛跑的参赛同学实在是太多了，连体育老师也忙不过来。这时体育老师发现了身为体育委员的苏大学神，便来找他帮忙。可是苏大学神需要热身，不然跑到一半就会抽（筋）、于是他就找到了你。。。如果你帮助体育老师解决了问题，老师就会给你5个积分。

题目描述

假设一共有N（2<=N<=20000）个参赛选手。（尼玛全校学生都没这么多吧）

老师会告诉你这N个选手的名字。

接着会告诉你M（1<=M<=1000000）句话，即告诉你学生A与学生B在同一个组里。

如果学生A与学生B在同一组里，学生B与学生C也在同一组里，就说明学生A与学生C在同一组。

然后老师会问你K（1<=K<=1000000）句话，即学生X和学生Y是否在同一组里。

若是则输出"Yes."，否则输出"No."

输入格式

第一行输入N和M。

接下来N行输入每一个同学的名字。

再往下M行每行输入两个名字，且保证这两个名字都在上面的N行中出现过，表示这两个参赛选手在同一个组里。

再来输入K。

接下来输入K个体育老师的询问。

输出格式

对于每一个体育老师的询问，输出"Yes."或"No."。

输入输出样例

输入 #1复制

10 6
Jack
Mike
ASDA
Michel
brabrabra
HeHe
HeHE
papapa
HeY
Obama
Jack Obama
HeHe HeHE
brabrabra HeHe
Obama ASDA
papapa Obama
Obama HeHE
3
Mike Obama
HeHE Jack
papapa brabrabra

输出 #1复制

No.
Yes.
Yes.

Solution

和并查集的模板题差不多

这次我还用了按秩合并！

以及

STL大法好！

Code

#include<iostream>
#include<algorithm>
#include<map>
using namespace std;
map<string,int>name;
int n,m,k,f[5001],size[5001],ans,yet;
int getf(int a)
{
    if(f[a]==a) return a;
    return f[a]=getf(f[a]);//路径压缩 
}
void merge(int a,int b)
{
    int i=getf(a);
    int j=getf(b);
    if(size[i]>size[j]) swap(i,j);
    f[j]=i;
    size[i]+=size[j];
}
int main()
{
    cin>>n>>m;
    string str,str1,str2;
    for(int i=1;i<=n;i++)
    {
        size[i]=1;
        f[i]=i;
    }
    for(int i=1;i<=n;i++)
    {
        cin>>str;
        name[str]=i;
    }
    for(int i=1;i<=m;i++){
        cin>>str1>>str2;
        merge(name[str1],name[str2]);
    }
    cin>>k;
    for(int i=1;i<=k;i++)
    {
        cin>>str1>>str2;
        if(getf(name[str1])==getf(name[str2])) cout<<"Yes.\n";
        else cout<<"No.\n";
    }
    return 0;
}