蒟蒻求教,3个TLE

回复帖子 返回题目

@ 智慧草 2017-08-13 17:23

代码如下

#include <iostream>
using namespace std;
int n,m;
int z,x,y; 
int fuck[10001];
int find(int x)
{
    if(fuck[x]!=x)
        x=find(fuck[x]);
    return x;
}
int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)
        fuck[i]=i;
    for(int i=1;i<=m;i++)
    {
        cin>>z>>x>>y;
        if(z==1)
        {
            int f1=find(x);
            int f2=find(y);
            if(f1!=f2)
            {
                fuck[f2]=f1;
                fuck[y]=f1;
            }
        }
        else
        {
            if(find(x)==find(y))
                cout<<"Y\n";
            else
                cout<<"N\n";
        }
    }
}
@ xun薰 2017-08-13 17:32 回复

find函数优化一下路径压缩就可以了

#include <iostream>
using namespace std;
int n,m;
int z,x,y; 
int fuck[10001];
int find(int x)
{
    return fuck[x]==x?x:fuck[x]=find(fuck[x]);
}
int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)
        fuck[i]=i;
    for(int i=1;i<=m;i++)
    {
        cin>>z>>x>>y;
        if(z==1)
        {
            int f1=find(x);
            int f2=find(y);
            if(f1!=f2)
            {
                fuck[f2]=f1;
                fuck[y]=f1;
            }
        }
        else
        {
            if(find(x)==find(y))
                cout<<"Y\n";
            else
                cout<<"N\n";
        }
    }
}