蓝桥杯 历届试题 合根植物 （并查集或dfs）

题目链接

 

题意：给定一个n*m的矩阵，里面有标号为1到n*m的植物，这些植物可以合并，题目将给出k对合并的植物，问合并以后，还有多少颗植物（意思是假如1和2合并了，那么就只算1颗）

思路：简而言之，就是求联通块的数量。那么有两种常用的方法

1.并查集

2.dfs

并查集很好理解，将能合并的连在一起就行。

dfs呢就是求联通块的数量。

反正很简单，直接写就行了。

反思：好久没写前向星了。

dfs代码：

#include <bits/stdc++.h>
using namespace std;
const int maxn=1e6+7;
int head[maxn];
int cnt;
bool vis[maxn];
struct Edge
{
    int u,v,next;
}edge[maxn*2];
void addEdge(int u,int v)
{
    edge[cnt].u=u;
    edge[cnt].v=v;
    edge[cnt].next=head[u];
    head[u]=cnt++;
}
void init()
{
    memset(head,-1,sizeof(head));
    memset(vis,1,sizeof(vis));
    cnt=0;
}
void dfs(int u)
{
    vis[u]=0;
    for(int i=head[u];i!=-1;i=edge[i].next)
    {
        int v=edge[i].v;
        if(v==u||vis[v]==0) continue;
        dfs(v);
    }
}
int main()
{
    #ifndef ONLINE_JUDGE
        freopen("in.txt","r",stdin);
        freopen("out.txt","w",stdout);
    #endif
    int n,m,k;
    init();
    scanf("%d%d%d",&n,&m,&k);
    for(int i=0,u,v;i<k;i++)
    {
        scanf("%d%d",&u,&v);
        addEdge(u,v);
        addEdge(v,u);
    }
    int ans=0;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
            if(vis[j+(i-1)*m])
            {
                dfs(j+(i-1)*m);
                ans++;
            }
        }
    }
    cout<<ans<<endl;
    return 0;
}

 

并查集代码：

#include <bits/stdc++.h>
using namespace std;
const int maxn=1e6+7;
int f[maxn];
int Find(int x)
{
    if(f[x]==x) return x;
    else return f[x]=Find(f[x]);
}
int main()
{
    #ifndef ONLINE_JUDGE
        freopen("in.txt","r",stdin);
        freopen("out.txt","w",stdout);
    #endif
    int n,m,k;
    scanf("%d%d%d",&n,&m,&k);
    for(int i=1;i<=n*m;i++)
    {
        f[i]=i;
    }
    for(int i=1,u,v;i<=k;i++)
    {
        scanf("%d%d",&u,&v);
        int fx=Find(u),fy=Find(v);
        if(fx!=fy)
        {
            f[fy]=fx;
        }
    }
    int ans=0;
    for(int i=1;i<=n*m;i++)
    {
        if(f[i]==i)
        {
            ans++;
        }
    }
    cout<<ans<<endl;
    return 0;
}