CodeVS 2639 约会计划

Portal：http://codevs.cn/problem/2639/

简单题，并查集再加个map，一开始我以为要用图论，失算

FML，我的代码怎么这么长。。。优化太多反而慢啊。。。大常数大师啊。。。

#include<iostream>
#include<algorithm>
#include<set>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<map>
#include<cstring>
using namespace std;
#define FOR(i,j,k) for(int i=j;i<=k;i++)
#define FORD(i,j,k) for(int i=j;i>=k;i--)
#define LL long long
#define maxn 2010
int father[maxn],rank[maxn];
int n,m,p;
string x,y;
map<string,int> name;
int setfind(int xx)
{
    int fa=father[xx];
    if(fa==xx) return fa;
    else return father[xx]=setfind(fa);
}
void setunion(int xx,int yy)
{
    int X=setfind(xx);
    int Y=setfind(yy);
    if(X==Y) return;
    if(rank[X]==rank[Y]) {father[X]=Y;rank[Y]++;}
    else if(rank[X]>rank[Y]) father[X]=Y;
    else father[Y]=X;
}
bool is_same(int xx,int yy)
{
    return setfind(xx)==setfind(yy);
}
int main()
{
cin>>n>>m>>p;
FOR(i,1,n)
{cin>>x;name[x]=i;father[i]=i;rank[i]=1;}
FOR(i,1,m)
{
    cin>>x>>y;
    setunion(name[x],name[y]);
}
FOR(i,1,p)
{
    cin>>x>>y;
    if(is_same(name[x],name[y])) cout<<"safe"<<endl;
    else cout<<"cc cry"<<endl;
}
return 0;
}

蛋疼地在路径压缩时做了个用空间换时间的优化

/*init*/i=1-m   rank[i]=1;
//
if(X==Y) return;
    if(rank[X]==rank[Y]) {father[X]=Y;rank[Y]++;}
    else if(rank[X]>rank[Y]) father[X]=Y;
    else father[Y]=X;
//平时并的时候都是直接father[X]=Y;，汗

UN3次才AC，粗心啊

啧，丢人啊