并查集的功能:
- 查询:两个元素是否在同一个集合中
- 合并:两个元素各自所在的集合合并
概念先不介绍了,网上资料很多
时间复杂度:
设数据量为N,那么只要查询与合并的次数整体逼近了O(N)及以上,平均每次的的时间复杂度为O(1)
练习题
这题正常做的话其实不难,递归就可以,主要思路是每次找到值为1的位置就开始递归,并将遍历到的值为1的位置置为2,这样做是防止下次又递归到上次的位置,代码如下:
public class Island {
private int count;
public void solution(int[][] arr) {
if (arr == null) return;
for (int i = 0; i < arr.length; i++) {
for (int j = 0; j < arr[0].length; j++) {
if (arr[i][j] == 1) {
count++;
infect(arr, i, j);
}
}
}
}
private void infect(int[][] arr, int i, int j) {
if (i < 0 || j <0 || i >= arr.length || j >= arr[0].length || arr[i][j]!=1) return;
arr[i][j] = 2;
infect(arr, i, j-1);
infect(arr, i, j+1);
infect(arr, i-1, j);
infect(arr, i+1, j);
}
public static void main(String[] args) {
int[][] arr = new int[][] {
{0,0,1,0,1,0},
{1,1,1,0,1,0},
{0,0,0,1,0,0},
{0,0,1,0,0,0}
};
Island island = new Island();
island.solution(arr);
System.out.println("岛的数量为: " + island.count);
}
}
看左神的算法课是,给出了一个讨论题:由于递归是单核的,所以当矩阵特别大时,使用多核是更好的。因此就想啊,能不能将矩阵分成多个子区域,每个区域分别由一个核计算,这样能大大提高程序运行速度。
但这里有个问题,分别计算多个小区域当然可以,但合并时怎么办?以下图为例:
- 假设现在将矩阵分成了两个区域分别求解,结果是左侧两个岛,右侧两个岛,但事实上只有两个岛。因此现在要解决的问题是如何处理边界。
- 使用并查集的思想,我们知道一个集合中只有一个代表节点,因此在每个子区域进行计算时将不同的岛弄成不同的并查集,对于左侧来说有两个集合,设为A和B;同理对于右侧来说,有两个集合,设为C和D:
- 现在处理边界,对于内部边界,如下图所示。A、C和B是可以合并的,并查集合并之后会变成一个大的集合,且集合中每个元素的代表节点都是同一个(对于下图来说,集合C和集合B的代表节点在合并后都是A),集合D无法合并:
- 合并后的并查集有两个,因此有两个岛。