本周学习了图论,最重要的是讲了两种搜索方法,分别是深度优先搜索和广度优先搜索。于是在leetcode上看到了这样一道很有意思的题目。数孤岛的数目。
先贴原题~
我的思路是:从第一个点开始便利整个矩阵,利用深度优先搜索,即找这个点的相邻的元素,查看是否为1,如果为1且没有被搜索过,就顺着这个方向一直找下去;如果该方向无法继续搜索,就换一个方向搜索,直至把四面的岛屿全部搜完。
贴出代码:
class Solution {
public:
void findRawAdd1(int i, int j, vector<vector<int>>& flag, vector<vector<char>>& grid) {
if (i + 1 < raw){
if (grid[i + 1][j] == '1' && flag[i + 1][j] == 0) {
flag[i + 1][j] = 1;
findRawAdd1(i + 1, j, flag, grid);
findColAdd1(i + 1, j, flag, grid);
findColMinus1(i + 1, j, flag, grid);
}
}
else return;
}
void findColAdd1(int i, int j, vector<vector<int>>& flag, vector<vector<char>>& grid) {
if (j + 1 < column){
if (grid[i][j + 1] == '1' && flag[i][j + 1] == 0) {
flag[i][j + 1] = 1;
findRawAdd1(i, j + 1, flag, grid);
findColAdd1(i, j + 1, flag, grid);
findRawMinus1(i, j + 1, flag, grid);
}
}
else return;
}
void findRawMinus1(int i, int j, vector<vector<int>>& flag, vector<vector<char>>& grid) {
if (i - 1 > -1) {
if (grid[i - 1][j] == '1' && flag[i - 1][j] == 0) {
flag[i - 1][j] = 1;
findColAdd1(i - 1, j, flag, grid);
findRawMinus1(i - 1, j, flag, grid);
findColMinus1(i - 1, j, flag, grid);
}
}
else return;
}
void findColMinus1(int i, int j, vector<vector<int>>& flag, vector<vector<char>>& grid) {
if (j - 1 > -1) {
if (grid[i][j - 1] == '1' && flag[i][j - 1] == 0) {
flag[i][j - 1] = 1;
findRawAdd1(i, j - 1, flag, grid);
findRawMinus1(i, j - 1, flag, grid);
findColMinus1(i, j - 1, flag, grid);
}
}
else return;
}
int numIslands(vector<vector<char>>& grid) {
result = 0;
if (!grid.empty()) {
raw = grid.size();
column = grid[0].size();
for (int j = 0; j < column; j++) {
line.push_back(0);
}
for (int i = 0; i < raw; i++) {
flag.push_back(line);
}
for (int i = 0; i < raw; i++) {
for (int j = 0; j < column; j++) {
if (flag[i][j] == 0 && grid[i][j] == '1') {
findRawAdd1(i, j, flag, grid);
findColAdd1(i, j, flag, grid);
result++;
}
}
}
}
return result;
}
private:
int raw, column, result;
vector<vector<int>> flag;
vector<int> line;
};
因为查找下一行的时候,是从上一行搜索过来的,所以findRawAdd1函数中不用调用findRawMinus1,其他的函数同理。
时间复杂度为:O(raw*column);
空间复杂度为:O(raw*column);即开了一个标记位数组。