UVALive - 7303 Aquarium(最小生成树)

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/dpdpd/article/details/52175945

主要就是建图的思路，然后套用一个最小生成树的模板
建图的思路：将线段划分的区域用并查集合并，然后对每个不同祖先的区域加到最小生成树里

#include<bits/stdc++.h>
using namespace std;
#define ll long long
char s[105][105];
int v[105][105];
int fa[200001];
int cnt;//线段的个数 
struct node
{
    int a,b;
}g[105][105];//格子的个数（每个格子分成两部分ab） 
struct node2
{
    int a,b,len;//线段左右的p值以及线段的权 
}edge[400005];
void add(int x,int y,int v)
{
    edge[cnt].a=x;//线段左值标为起点
    edge[cnt].b=y;//线段右值标为终点
    edge[cnt++].len=v;
}
bool cmp(node2 x,node2 y)
{
    return x.len<y.len;
}
int findfa(int a)
{
    return fa[a]==a?fa[a]:fa[a]=findfa(fa[a]);//不同祖先的归到一个祖先上
}
void join(int a,int b)
{
    int faa=findfa(a),fbb=findfa(b);
    if(faa!=fbb)
    fa[faa]=fbb;
}
int main()
{
    int t;
    cin>>t;
    for(int z=1;z<=t;z++)
    {
        memset(s,0,sizeof(s));      
        cnt=0;
        int r,c;
        cin>>c>>r;
        for(int i=0;i<c;i++)
        cin>>s[i];
        for(int i=0;i<c;i++)
        {
            for(int j=0;j<r;j++)
                cin>>v[i][j];
        }
        int p=1;
        for(int i=0;i<c;i++)
        {
            for(int j=0;j<r;j++)
            {
                if(j==0)
                {
                    g[i][j].a=p++;
                    g[i][j].b=p++;
                    add(g[i][j].a,g[i][j].b,v[i][j]); 
                }
                else
                {
                    g[i][j].a=g[i][j-1].b;
                    g[i][j].b=p++;
                    add(g[i][j].a,g[i][j].b,v[i][j]);
                }
                if(i!=0)
                {
                    if(s[i-1][j]=='/')
                    {
                        if(s[i][j]=='/')
                        {
                            add(g[i][j].a,g[i-1][j].b,0);//将格子的联通部分合起来
                        }
                        else
                        {
                            add(g[i][j].b,g[i-1][j].b,0);
                        }
                    }
                    else
                    {
                        if(s[i][j]=='/')
                        {
                            add(g[i][j].a,g[i-1][j].a,0);
                        }
                        else
                        {
                            add(g[i][j].b,g[i-1][j].a,0);
                        }
                    }
                }
            }
        }
        for(int i=1;i<=200001;i++)
        {
            fa[i]=i;
        }
        sort(edge,edge+cnt,cmp);
        int ans=0;
        for(int i=0;i<cnt;i++) 
        {
            int u=edge[i].a;
            int v=edge[i].b;
            if(findfa(u)!=findfa(v))
            {
                join(u,v);
                ans+=edge[i].len;
            } 
        }
        printf("Case %d: %d\n",z,ans);
    }
}