LeetCode周赛第206场 5513. 连接所有点的最小费用

在这里插入图片描述

思路:

  • 首先思路就是通过最小生成树的思想去将所有点连接起来,这样就能保证我们的得到的是最小费用
  • 我们把点存在一个边的结构体中,通过一个二重循环将所有的边保存下来,然后再用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;
    }
};

猜你喜欢

转载自blog.csdn.net/qq_43663263/article/details/108562771