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

2021-10-14 16:31:41 阅读：161 来源： 互联网

标签：return ve 1144 get int d% 最小格点 find

在这里插入图片描述
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;
}

标签：return,ve,1144,get,int,d%,最小,格点,find
来源： https://blog.csdn.net/bettle_king/article/details/120766882

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

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