547. Friend Circles【用warshall算法求解】

来源:互联网 发布:社会化传播矩阵 编辑:程序博客网 时间:2024/06/10 21:22

https://leetcode.com/problems/friend-circles/#/description

这个题目,给出一个N*N的矩阵M,如果M[i][j]==1,说明i和j有直接朋友关系,如果M[i][j]==1并且M[j][k]==1那么i和k就有间接朋友关系。求所有人之间的朋友圈子有几个。

一开始的想法是用warshall算法,计算闭包,那么有多少个链子应该能很快计算出来。后来,闭包是算出来了,有多少个链子还没算出来……以后再补充吧……

warshall算法:http://www.th7.cn/Program/cp/201606/876462.shtml


****************************************warshall分界线************************************************************

public int findCircleNum(int[][] M) {


int i, j, k;

//计算闭包
for (i = 0; i < M.length; i++) {
for (j = 0; j < M.length; j++) {
for (k = 0; k < M.length; k++) {
if (M[j][i] != 0)
M[j][k] = M[j][k] + M[i][k];
if (M[j][k] != 0)
M[j][k] = 1;
}
}
}


Map<Integer, Integer> map = new HashMap<Integer, Integer>();
int total = 0;
 
for (i = 0; i < M.length; i++) {
if (map.containsKey(i)) {// 已经遍历过,跳过
continue;
} else {// 未遍历
for (j = 0; j < M.length; j++) {
if (M[i][j] == 1) {
if (!map.containsKey(j)) {// 如果还未遍历,则存入map
map.put(j, 1);
}
}


}
total++; //每多遍历一行,说明环多一个
}
}


return total;
}

后来看了一下讨论,原来还有一个熟悉的算法叫做并查集。http://blog.csdn.net/dm_vincent/article/details/7655764

还没做……

然后也可以用dfs做……回来再慢慢看吧……


warshall算法:http://www.th7.cn/Program/cp/201606/876462.shtml
0 0