leetcode刷题笔记547

写在前边

昨天实在太忙了没有做题，我看了一眼也没有什么很好的思路，就没去想，也没去做。今天本来也没打算做题，但是看了一眼今天的题目，发现是很直接的一道题，就是考察并查集的用法的，刚巧数据结构项目中要求实现Kruskal算法，而这个算法中就要求使用并查集。所以这道题算是真正意义上的秒杀，从开始做到解决没有用20分钟。

题目

有 n 个城市，其中一些彼此相连，另一些没有相连。如果城市 a 与城市 b 直接相连，且城市 b 与城市 c 直接相连，那么城市 a 与城市 c 间接相连。

省份 是一组直接或间接相连的城市，组内不含其他没有相连的城市。

给你一个 n x n 的矩阵 isConnected ，其中 isConnected[i][j] = 1 表示第 i 个城市和第 j 个城市直接相连，而 isConnected[i][j] = 0 表示二者不直接相连。

返回矩阵中 省份 的数量。

示例 1：

输入：isConnected = [[1,1,0],[1,1,0],[0,0,1]]
输出：2
示例 2：

输入：isConnected = [[1,0,0],[0,1,0],[0,0,1]]
输出：3

提示：

1 <= n <= 200
n == isConnected.length
n == isConnected[i].length
isConnected[i][j] 为 1 或 0
isConnected[i][i] == 1
isConnected[i][j] == isConnected[j][i]

就是给出一所有城市彼此之间直接相连的情况，求出有多少个省份。所有连在一起的城市，不论是直接相连还是间接相连，都算是在同一个“省份”中。

这很直接的就是考并查集，只需要把所有的连通分量合并起来，用数组表示一下，其中负数表示该分量的个数，正数代表父节点，这样一棵棵子树构成并查集。先假设所有城市在不同的省份，即省份数=n,每挂上一个城市，令n自减，这样就可以最终得到省份的数量。由于是自己书写的数据结构，速度很快，但是空间开销有点大。。。

源码如下：

#include"iostream"
#include"vector"
using namespace std;

class Solution {
public:
    void Unoin(int i,int j,vector<int>& list,int& count)
    {
        while(list[i]>0)
        {
            i=list[i];
        }
        while(list[j]>0)
        {
            j=list[j];
        }
        if(i==j)
        {
            return;
        }
        else
        {
            if(list[i]<list[j])
            {
                list[j]=i;
                list[i]--;
            }
            else
            {
                list[i]=j;
                list[j]--;
            }
            count--;
            return;  
        }
    }
    int UFsets(vector<vector<int>> temp)
    {
        int num=temp.size();
        int count=num;
        vector<int> list(num);
        for(int i=0;i<num;i++)
        {
            list[i]=-1;
        }
        for(int i=0;i<num;i++)
        {
            for(int j=0;j<temp[i].size();j++)
            {
                if(temp[i][j]==1)
                {
                    if(i!=j)
                    {
                        Unoin(i,j,list,count);
                    }
                }
            }
        }

        return count;
         
    }
    int findCircleNum(vector<vector<int>>& isConnected) {
       return UFsets(isConnected);
    }
};
int main()
{
    vector<vector<int>> v;
    v.push_back({1});
   /* v.push_back({0,1,1,1});
    v.push_back({1,1,1,0});
    v.push_back({0,1,0,1});*/
    Solution s;
    cout<<s.findCircleNum(v)<<endl;
    system("pause");
    return 0;
}

main()函数中是测试案例，怕submit的时候有莫名其妙的错误。。。