1144. 连接格点【最小生成树】

1144. 连接格点【最小生成树】
https://www.acwing.com/problem/content/description/1146/

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+10;
int p[N],n,m;
int x,y,xx,yy;
struct node{int a,b,c;};
bool cmp(node a,node b) {return a.c<b.c;}
vector<node>ve;
int find(int x)
{
    if(x!=p[x]) p[x]=find(p[x]);
    return p[x];
}
int get(int x,int y)
{
    return x*m+y;
}
void solve(int x,int y)//存所有的边
{
    int dx[4]={-1,0,1,0};
    int dy[4]={0,1,0,-1};
    for(int i=0;i<4;i++)
    {
        int tempx=x+dx[i];
        int tempy=y+dy[i];
        if(tempx<1||tempx>n||tempy<1||tempy>m) continue;
        if(get(x-1,y)<get(tempx-1,tempy))//a<b时才加可以避免重复加
        {
            if(dy[i]) ve.push_back({get(x-1,y),get(tempx-1,tempy),2});
            else ve.push_back({get(x-1,y),get(tempx-1,tempy),1});
        }
    }
}
void init()//初始化
{
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++) p[get(i-1,j)]=get(i-1,j);
}
int f()
{
    int res=0;
    sort(ve.begin(),ve.end(),cmp);
    for(int i=0;i<ve.size();i++)
    {
        int a=ve[i].a,b=ve[i].b,c=ve[i].c;
        if(find(a)!=find(b)) 
        {
            p[find(a)]=find(b);
            res+=c;
        }
    }
    return res;
}
int main(void)
{
    scanf("%d%d",&n,&m);
    init();
    while(scanf("%d%d%d%d",&x,&y,&xx,&yy)!=EOF)
    {
        int a=get(x-1,y),b=get(xx-1,yy);
        if(find(a)!=find(b)) p[find(a)]=find(b);
    }
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++) solve(i,j);
    printf("%d\n",f());
    return 0;
}

因为边权只有1和2两种情况,故可以开俩个数组,这样就不用排序了。减少时间。

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+10;
int p[N],n,m;
int x,y,xx,yy;
struct node{int a,b,c;};
bool cmp(node a,node b) {return a.c<b.c;}
vector<node>ve1,ve2;
int find(int x)
{
    if(x!=p[x]) p[x]=find(p[x]);
    return p[x];
}
int get(int x,int y)
{
    return x*m+y;
}
void solve(int x,int y)//存所有的边
{
    int dx[4]={-1,0,1,0};
    int dy[4]={0,1,0,-1};
    for(int i=0;i<4;i++)
    {
        int tempx=x+dx[i];
        int tempy=y+dy[i];
        if(tempx<1||tempx>n||tempy<1||tempy>m) continue;
        if(get(x-1,y)<get(tempx-1,tempy))//a<b时才加可以避免重复加
        {
            if(dy[i]) ve2.push_back({get(x-1,y),get(tempx-1,tempy),2});
            else ve1.push_back({get(x-1,y),get(tempx-1,tempy),1});
        }
    }
}
void init()//初始化
{
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++) p[get(i-1,j)]=get(i-1,j);
}
int f()
{
    int res=0;
    //sort(ve.begin(),ve.end(),cmp);
    for(int i=0;i<ve1.size();i++)
    {
        int a=ve1[i].a,b=ve1[i].b,c=ve1[i].c;
        if(find(a)!=find(b)) 
        {
            p[find(a)]=find(b);
            res+=c;
        }
    }
    for(int i=0;i<ve2.size();i++)
    {
        int a=ve2[i].a,b=ve2[i].b,c=ve2[i].c;
        if(find(a)!=find(b)) 
        {
            p[find(a)]=find(b);
            res+=c;
        }
    }
    return res;
}
int main(void)
{
    scanf("%d%d",&n,&m);
    init();
    while(scanf("%d%d%d%d",&x,&y,&xx,&yy)!=EOF)
    {
        int a=get(x-1,y),b=get(xx-1,yy);
        if(find(a)!=find(b)) p[find(a)]=find(b);
    }
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++) solve(i,j);
    printf("%d\n",f());
    return 0;
}
上一篇:怒肝2W长文 !带你进入数据仓库Hive的世界【理论+实践】


下一篇:熬夜总结了面试套路,2W字长文!