思路:
- 首先思路就是通过最小生成树的思想去将所有点连接起来,这样就能保证我们的得到的是最小费用
- 我们把点存在一个边的结构体中,通过一个二重循环将所有的边保存下来,然后再用sort函数将其从小到大排序
- 接下来就是用并查集依次将不同的边连接起来,然后最终能够将点连在一起,这时得到的就是最小费用。
- 代码用的是kruskal的思想,时间复杂度还是有点高的
代码:
class Solution {
public:
struct Edge
{
int u,v,w;
}edge[1000005];
int fa[1000005],n,m,ans,eu,ev,cnt;
int find(int x)
{
while(x!=fa[x]) x=fa[x]=fa[fa[x]];
return x;
}
static bool cmp(Edge a,Edge b)
{
return a.w<b.w;
}
void kruskal()
{
sort(edge,edge+m,cmp);
for(int i=0;i<m;i++)
{
eu=find(edge[i].u), ev=find(edge[i].v);
if(eu==ev)
{
continue;
}
ans+=edge[i].w;
fa[ev]=eu;
if(++cnt==n-1)
{
break;
}
}
}
int minCostConnectPoints(vector<vector<int>>& points) {
for(int i=0;i<1000005;i++)
{
fa[i]=i;
}
for(int i=0;i<points.size();i++)
{
for(int j=0;j<points.size();j++)
{
if(i!=j)
{
int dis=abs(points[i][0]-points[j][0])+abs(points[i][1]-points[j][1]);
edge[m].u=i,edge[m].v=j,edge[m].w=dis;
m++;
}
}
}
n=points.size();
kruskal();
return ans;
}
};