LeetCode - python - 547. 모멘트

반 에 있다 N 학생그 중 어떤 사람 은 친구 이 고, 어떤 사람 은 그렇지 않다.그들의 우정 은 전달 성 을 가지 고 있다.하면, 만약, 만약... 친구 의 친구, 그러면 우 리 는 A 도 C 라 고 생각 할 수 있다. 친구이른바 친구 권 이란 모든 친구 의 집합 을 가리킨다.
하 나 를 정 하 다 N * N 행렬 M. 반 중 학생 간 의 친구 관 계 를 나타 낸다.만약 에 M [i] [j] = 1 은 i 번 째 와 j 번 째 학생 이 서로 친구 가 되 는 것 을 알 고 있 음 을 나타 낸다. 그렇지 않 으 면 모른다.너 는 모든 학생 들 이 이미 알 고 있 는 친구 권 의 총 수 를 출력 해 야 한다.
예시 1:

  : 
[[1,1,0],
 [1,1,0],
 [0,0,1]]
  : 2 
  ：    0   1    ，        。
 2           。    2。

예시 2:

  : 
[[1,1,0],
 [1,1,1],
 [0,1,1]]
  : 1
  ：    0   1    ，  1   2    ，    0   2    ，            ，  1。

주의:
N 은 [1, 200] 의 범위 내 에 있다
4. 567917. 모든 학생 에 게 M [i] [i] = 1 이 있 습 니 다
4. 567917. M [i] [j] = 1 이 있 으 면 M [j] [i] = 1 이 있다
첫 번 째 사고방식:
이용 하고 수집 하여 해결 하 다.

class UnionFindSet(object):
    def __init__(self, grid):

        m, n = len(grid), len(grid[0])
        self.roots = [i for i in range(m)]
        self.rank = [0 for i in range(m)]
        self.count = n
        
    def find(self, member):
        tmp = []
        while member != self.roots[member]:
            tmp.append(member)
            member = self.roots[member]
        for root in tmp:
            self.roots[root] = member
        return member
        
    def union(self, p, q):
        parentP = self.find(p)
        parentQ = self.find(q)
        if parentP != parentQ:
            if self.rank[parentP] > self.rank[parentQ]:
                self.roots[parentQ] = parentP
            elif self.rank[parentP] < self.rank[parentQ]:
                self.roots[parentP] = parentQ
            else:
                self.roots[parentQ] = parentP
                self.rank[parentP] -= 1
            self.count -= 1
 
class Solution(object):
    def findCircleNum(self, M):
        """
        :type M: List[List[int]]
        :rtype: int
        """
        if not M or not M[0]:
            return 0
        
        ufs = UnionFindSet(M)
        
        for i in range(len(M)):
            for j in range(len(M[0])):
                if M[i][j] == 1:
                    ufs.union(i, j) 
                    
        return ufs.count