leetcode 1135. 最低成本联通所有城市（C++）

想象一下你是个城市基建规划者，地图上有 N 座城市，它们按以 1 到 N 的次序编号。

给你一些可连接的选项 conections，其中每个选项 conections[i] = [city1, city2, cost] 表示将城市 city1 和城市 city2 连接所要的成本。（连接是双向的，也就是说城市 city1 和城市 city2 相连也同样意味着城市 city2 和城市 city1 相连）。

返回使得每对城市间都存在将它们连接在一起的连通路径（可能长度为 1 的）最小成本。该最小成本应该是所用全部连接代价的综合。如果根据已知条件无法完成该项任务，则请你返回 -1。

 

示例 1：

输入：N = 3, conections = [[1,2,5],[1,3,6],[2,3,1]]
输出：6
解释：
选出任意 2 条边都可以连接所有城市，我们从中选取成本最小的 2 条。

示例 2：

输入：N = 4, conections = [[1,2,3],[3,4,4]]
输出：-1
解释： 
即使连通所有的边，也无法连接所有城市。

 

提示：

1. 1 <= N <= 10000
2. 1 <= conections.length <= 10000
3. 1 <= conections[i][0], conections[i][1] <= N
4. 0 <= conections[i][2] <= 10^5
5. conections[i][0] != conections[i][1]

C++

class Solution {
public:
    static bool cmp(vector<int>& A, vector<int>& B)
    {
        return A[2]<B[2];
    }
    
    int func(int val, vector<int>& dp)
    {
        if(dp[val]==val)
        {
            return val;
        }
        else
        {
            return func(dp[val],dp);
        }
    }
    
    int minimumCost(int N, vector<vector<int>>& conections) 
    {
        int n=conections.size();
        sort(conections.begin(),conections.end(),cmp);
        int res=0;
        vector<int> dp(N+1,0);
        for(int i=1;i<N;i++)
        {
            dp[i]=i;    //初始状态，只跟自身相连
        }
        for(int i=0;i<n;i++)
        {
            int left=conections[i][0];
            int right=conections[i][1];
            int distance=conections[i][2];
            int d1=func(left,dp);
            int d2=func(right,dp);
            if(d1!=d2)
            {
                res+=distance;
                dp[d1]=d2;
            }
        }
        for(int i=2;i<=N;i++)
        {
            if(func(i,dp)!=func(i-1,dp))
            {
                return -1;
            }
        }
        return res;
    }
};