一、题目
班上有 N 名学生。其中有些人是朋友,有些则不是。他们的友谊具有是传递性。如果已知 A 是 B 的朋友,B 是 C 的朋友,那么我们可以认为 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 。
提示:
- 1 <= N <= 200
- M[i][i] == 1
- M[i][j] == M[j][i]
二、解决
朋友圈二维数组表示满足下面基本性质:
- n n n个人,最多有 n n n个朋友圈。
- i 、 j i、j i、j 为朋友,则对应二维数组记录中,点 M ( i , j ) M(i,j) M(i,j)为1,则 M ( j , i ) M(j,i) M(j,i)也为1,满足对称性。
- M{ {i,1}, {i,2}, …, {i,j+2}…{i,n}} 代表第 i 个人与这n个人的关系。这是一个朋友圈,当然其中某些值0,说明i与这些人不存在关系。但是i与剩下的几个人共同组成了一个朋友圈。如果这行值全为1,则只有一个朋友圈。
DFS与BFS的基本含义可看文章朋友圈进行理解。
1、DFS
思路:
每一行代表一个人,先从第一行开始,查找与(0,0)朋友关系的人(第1个人),如果有,则 (0, j)为1,在位置为第 1 行的第 j+1 列,即第 j+1 个人与第一个人有朋友关系。此时,继续往下递归,即到 第 j+1 行中,查找该人有朋友关系的人。期间遇到所有为1的人组成了一个朋友圈。
过程:
S1:逐行遍历二维矩阵;
S2:在这一行对每个节点进行访问判断,对于遇到值为1且未访问的节点,标记为已经访问;
S3:然后继续往下进行深度递归遍历。(将该行及对称节点标记为已经访问,因为这是一个朋友圈,结合基本性质2、3理解。)
代码:
class Solution {
public int findCircleNum(int[][] M) {
int[] visited = new int[M.length]; // 基本性质1
int count = 0;
for (int i=0; i<M.length; i++) {
if (visited[i]==0) {