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;
}