LeetCode 200. Number of Islands (并查集)
2017-04-30 18:42
357 查看
LeetCode 200. Number of Islands (并查集)
LeetCode 200 Number of Islands 并查集问题描述
解题思路
参考代码
Tags:
* Depth-first Search
* Breadth-first Search
* Union Find
问题描述
Given a 2d grid map of ‘1’s (land) and ‘0’s (water), count the number of islands. An island is surrounded by water and is formed by connecting adjacent lands horizontally or vertically. You may assume all four edges of the grid are all surrounded by water.Example 1:
11110 11010 11000 00000
Answer: 1
Example 2:
11000 11000 00100 00011
Answer: 3
解题思路
问题意思是:给一个用0和1组成的2维矩阵,0表示水,1表示陆地。被水包围的陆地就是岛。
1(陆地)相连组成岛,只能垂直或者水平相连,不能对角相连。
要求返回矩阵中岛的个数。
可以使用并查集,将相连的1都连成一个部分,最后得出部分总数就是岛的数量。
参考代码
/*
Given a 2d grid map of '1's (land) and '0's (water), count the number of islands. An island is surrounded by water and is formed by connecting adjacent lands horizontally or vertically. You may assume all four edges of the grid are all surrounded by water.
Example 1:
11110 11010 11000 00000
Answer: 1
Example 2:
11000 11000 00100 00011
Answer: 3
*/
#include <iostream>
#include <vector>
using namespace std;
class UF
{
public:
int *id;
int *sz;
int count;
public:
UF(int N)
{
id = new int
;
sz = new int
;
count = N;
for (int i = 0; i < N; ++i)
{
id[i] = i;
sz[i] = 1;
}
}
~UF()
{
delete[] id;
delete[] sz;
}
int find(int i)
{
while(id[i] != i)
{
id[i] = id[id[i]];
i = id[i];
}
return i;
}
bool connected(int p, int q)
{
return find(p) == find(q);
}
void connect(int p, int q)
{
int a = find(p);
int b = find(q);
if (a == b) return;
if(sz[a] < sz[b])
{
id[a] = b;
sz[b] += sz[a];
}
else
{
id[b] = a;
sz[a] += sz[b];
}
--count;
}
};
class Solution {
public:
int numIslands(vector<vector<char>>& grid) {
int zeroCount = 0; // number of water grid
int row = grid.size();
if (row == 0)return 0;
int col = grid[0].size();
UF uf = UF(row * col);
for (int i = 0; i < row; ++i)
{
for (int j = 0; j < col; ++j)
{
cout << "row = " <<i << ", col = " << j << endl;
if(grid[i][j] == '1')
{
if (i - 1 >= 0 && grid[i - 1][j] == '1')
uf.connect(i * col + j, (i - 1) * col + j);
if (i + 1 < row && grid[i + 1][j] == '1')
uf.connect(i * col + j, (i + 1) * col + j);
if (j - 1 >= 0 && grid[i][j - 1] == '1')
uf.connect(i * col + j, i * col + j - 1);
if (j + 1 < col && grid[i][j + 1] == '1')
uf.connect(i * col + j, i * col + j + 1);
}
else
++zeroCount;
}
}
return uf.count - zeroCount;
}
};
int main()
{
vector<vector<char>> grid =
{
{ '1', '1', '0', '0', '0' },
{ '1', '1', '0', '0', '0' },
{ '0', '0', '1', '0', '0' },
{ '0', '0', '0', '1', '1' }
};
auto sl = new Solution();
cout << sl->numIslands(grid) << endl;
system("pause");
return 0;
}
相关文章推荐
- Leetcode 200. Number of Islands
- <LeetCode OJ> 200. Number of Islands
- Leetcode 200. Number of Islands & 305. Number of Islands II
- 【LeetCode】200. Number of Islands
- LeetCode--200. Number of Islands
- LeetCode 200. Number of Islands
- LeetCode 200. Number of Islands(小岛)
- DFS求岛的个数LeetCode 200. Number of Islands
- LeetCode 200. Number of Islands
- leetcode 200. Number of Islands(DFS)
- leetcode解题之200. Number of Islands Java版(岛屿的数量)
- LeetCode *** 200. Number of Islands (Depth-first Search)
- [LeetCode] 200. Number of Islands 解题思路
- leetcode 200. Number of Islands DFS
- [LeetCode] 200. Number of Islands
- 【leetcode】200. Number of Islands
- leetcode200. Number of Islands
- LeetCode 200. Number of Islands
- leetcode 200. Number of Islands
- LeetCode 200. Number of Islands