iOS 面试题(14):计算有多少个岛屿

注:本文摘自唐巧博客,方便以后查阅。请谅解

今天这篇是算法系列面试题的最后一篇了,之后的面试题我将继续选择 iOS 开发相关的一些问题来讨论。

问题

在一个地图中,找出一共有多少个岛屿。

我们用一个二维数组表示这个地图,地图中的 1 表示陆地,0 表示水域。一个岛屿是指由上下左右相连的陆地,并且被水域包围的区域。

你可以假设地图的四周都是水域。

例一:

11110
11010
11000
00000

以上地图,一共有 1 个岛屿。

例二:

11000
11000
00100
00011

以上地图,一共有 3 个岛屿。

答案

这是 LeetCode 上的 第 200 题,我们可以用一种被称为「种子填充」(floodfill)的办法来解决此题。

具体的做法是:

1、遍历整个地图,找到一个未被标记过的,值为 1 的坐标。
2、从这个坐标开始,从上下左右四个方向,标记相邻的 1 。
3、把这些相邻的坐标,都标记下来,递归的进行标记,以便把相邻的相邻块也能标记上。
4、待标记全部完成之后,将岛屿的计数 +1。
5、回到第 1 步。如果第 1 步无法找到未标记的坐标,则结束。

虽然思路简单,但是实现起来代码量也不算小。这里有一些小技巧:

1、我们可以将上下左右四个方向的偏移量保存在数组中,这样在计算位置的时候,写起来更简单一些。
2、递归的标记过程可以用深度优先搜索(DFS)或者宽度优先搜索(BFS)。
以下是完整的参考代码:

private class Solution {
    private var flag: [[Int]]
    private var answer: Int
    private var movex : [Int] {        return [-1, 1, 0, 0]
    }
    private var movey : [Int] {        return [0, 0, -1, 1]
    }    init() {
        flag = [[Int]]()
        answer = 0
    }    func dfs(_ grid: [[Character]] ,_ x: Int,_ y: Int) {     
                        for i in 0..<4 {            
                         let tox = x + movex[i]          
                         let toy = y + movey[i]           
                         if tox >= 0 && tox < grid.count && toy >= 0
                           && toy < grid[0].count && grid[tox][toy] == "1" 
                           && flag[tox][toy] == 0 {
                                         flag[tox][toy] = 1
                                        dfs(grid, tox, toy)
            }
        }
    }    func numIslands(_ grid: [[Character]]) -> Int {
                        answer = 0
        flag = grid.map { $0.map { _ in return 0 }}      
                     for i in 0..<grid.count {          
                            for j in 0..<grid[i].count {            
                                if grid[i][j] == "1" && flag[i][j] == 0 {
                                         flag[i][j] = 1
                                         // print("find in \(i), \(j)")
                                        dfs(grid, i, j)
                                       answer += 1
                               }
                          }
            }       
       return answer
    }
}

Swift 的参数默认是不能修改值的,但是如果是 C++ 语言的话,我们可以直接在地图上做标记。因为地图只有 0 和 1 两种值,我们可以用 2 表示「标记过的陆地」,这样就省略了额外的标记数组。以下是我写的一个 C++ 的示例程序:

class Solution {
public:
    void fillLands(vector<vector<char>>& grid, int px, int py) {
        int movex[] = {0, 0, 1, -1};
        int movey[] = {-1, 1, 0, 0};
        queue<pair<int, int>> q;
        q.push(make_pair(px, py));
        grid[px][py] = '2';
        while (!q.empty()) {
            pair<int, int> item = q.front();
            q.pop();
            int tox, toy;
            for (int i = 0; i < 4; ++i) {
                tox = item.first + movex[i];
                toy = item.second + movey[i];
                if (tox >= 0 && tox < grid.size()
                    && toy >=0 && toy < grid[0].size()
                    && grid[tox][toy] == '1') {
                    grid[tox][toy] = '2';
                    q.push(make_pair(tox, toy));
                }
            }
        }
    }
    int numIslands(vector<vector<char>>& grid) {
        int ans = 0;
        for (int i = 0; i < grid.size(); ++i) {
            for (int j = 0; j < grid[0].size(); ++j) {
                if (grid[i][j] == '1') {
                    fillLands(grid, i, j);
                    ans++;
                }
            }
        }
        return ans;
    }
};

微信公众号编辑器对代码排版支持很差,更好的代码排版,请看:https://gist.github.com/tangqiaoboy/5a5e5beb2cd020ed1f803958a8f2c318

这类题目有相当多的变种,希望大家能够掌握。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 214,504评论 6 496
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,434评论 3 389
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 160,089评论 0 349
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,378评论 1 288
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,472评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,506评论 1 292
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,519评论 3 413
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,292评论 0 270
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,738评论 1 307
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,022评论 2 329
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,194评论 1 342
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,873评论 5 338
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,536评论 3 322
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,162评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,413评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,075评论 2 365
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,080评论 2 352

推荐阅读更多精彩内容

  • 问题:在一个地图中,找出一共有多少个岛屿。 我们用一个二维数组表示这个地图,地图中的 1 表示陆地,0 表示水域。...
    凯旋之歌阅读 587评论 0 0
  • http://blog.csdn.net/david21984/article/details/57451917 ...
    紫色冰雨阅读 554评论 0 0
  • Spring Cloud为开发人员提供了快速构建分布式系统中一些常见模式的工具(例如配置管理,服务发现,断路器,智...
    卡卡罗2017阅读 134,646评论 18 139
  • 愚人节表白的人 都是想爱想疯了 我的丘比特啊 求求你 一箭射死认真的人。
    留子尧阅读 377评论 2 7
  • 神奇的周末,居然比平时上班醒的还早!原本打算窝在被窝里再懒一会儿的,可小家伙却不答应呢,左踢右踹的。索性起来收拾停...
    18姨的自留地阅读 122评论 0 0