Lintcode解题笔记 - 岛屿的个数

原文链接: http://www.lintcode.com/zh-cn/problem/number-of-islands/

给一个01矩阵,求不同的岛屿的个数。

0代表海,1代表岛,如果两个1相邻,那么这两个1属于同一个岛。我们只考虑上下左右为相邻。

样例
在矩阵:

[
[1, 1, 0, 0, 0],
[0, 1, 0, 0, 1],
[0, 0, 0, 1, 1],
[0, 0, 0, 0, 0],
[0, 0, 0, 0, 1]
]
中有 3 个岛.

思路一:
使用一个辅助数组,visited[][],大小和原数组gird相同,遍历grid数组,如果对应元素grid[i][j]在visited中也没有访问,则将visited[i][j]设置为1, 然后遍历visited[i][j]的上下左右4个元素。

使用辅助的floodRush函数,该函数用于判断/设定visited[i][j]的值,然后递归遍历visited[i][j]上下左右4个元素

该解法在第8个case时候,遇到超时问题,可能是时间复杂度过高

class Solution {
public:
    /**
     * @param grid a boolean 2D matrix
     * @return an integer
     */
    int numIslands(vector<vector<bool>>& grid) {
        // Write your code here
        int row = grid.size();
        if(row < 1)
            return 0;
        int column = grid[0].size();
        vector<int> tmp(column, 0);
        vector<vector<int>> visited(row, tmp); 
        //int visited[row][column] = 0;
        //vector<vector<bool>> visited = grid;
        int result = 0;
        for(int i = 0; i < row; i++){
            for(int j = 0; j < column; j++ ){
                if(grid[i][j] == true && visited[i][j] != 1){
                    floodRush(i,j, visited, grid);
                    result++;
                }

            }
        }
        return result;



    }


    void floodRush(int row, int column, vector<vector<int>> & visited, vector<vector<bool>> grid){
        if(row < 0 || column < 0 || row >= grid.size() || column >= grid[0].size() ){
            return;
        }
        if(visited[row][column] == 1)
            return;
        if(grid[row][column] == true){
            visited[row][column] = 1;
            floodRush(row-1, column, visited, grid);
            floodRush(row+1, column, visited, grid);
            floodRush(row, column-1, visited, grid);
            floodRush(row, column+1, visited, grid);
        }else{
            return;
        }



    }

};

几个总结,传递以为数组时候,可以使用void A(int a[]),但传递二维数组的时候,需要使用A(int a[][xxx]),来传递。因为编译器需要之后二维数组中某一维度的深度。 参见http://www.cplusplus.com/doc/tutorial/arrays/

阅读更多
文章标签: lintcode
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

不良信息举报

Lintcode解题笔记 - 岛屿的个数

最多只允许输入30个字

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭