LeetCode:岛屿的个数(Swift)

最近对算法突然来了兴趣,奈何大学学的数据结构算法已经丢完了,只有自己一点点的捡回来啰。
言归正传,来看看LeetCode一道经典题:岛屿的个数
此题可以使用BFS,DFS来解就可以了,我这里分别用了栈,队列,递归三种方法来解,上代码。

class Solution {
    

    var rows : Int!
    var cols : Int!
    var visited : [[Bool]]!
    var stack : [CGPoint] = []

    func numIslands(_ grid : [[Character]]) -> Int {
        if (grid.count == 0){
            return 0
        }
        var ans = 0
        rows = grid.count
        cols = grid[0].count
        visited = Array(repeating: Array(repeating: false, count: cols), count: rows)
        for i in 0..<rows {
            for j in 0..<cols{
                if (grid[i][j] == "1" && visited[i][j] == false ){
                    ans += 1
                    //BFS(getBFS: grid, i: i, j: j)    //递归
                    //BFSWithQueue(grid, i: i, j: j) //队列
                    //stack = []
                    //DFSWithStack(grid, i: i, j: j) //栈
                }
                
            }
        }
        
        return ans
    }

    func BFS(getBFS grid : [[Character]], i : Int, j : Int){
        if(i>=0 && i<rows && j>=0 && j<cols){
            if(grid[i][j] == "1" && visited[i][j] == false){
                visited[i][j] = true
                BFS(getBFS: grid, i: i+1, j: j)
                BFS(getBFS: grid, i: i-1, j: j)
                BFS(getBFS: grid, i: i, j: j-1)
                BFS(getBFS: grid, i: i, j: j+1)
            }
        }
    }
    
    func BFSWithQueue(_ grid : [[Character]], i : Int, j : Int) {
        var queue : [CGPoint] = []
        queue.append(CGPoint.init(x: i, y: j))
        while !queue.isEmpty {
            let x = Int(queue.first!.x)
            let y = Int(queue.first!.y)

            if(x>=0 && x<rows && y>=0 && y<cols){
                if(grid[x][y] == "1" && visited[x][y] == false){
                    queue.append(CGPoint.init(x: x+1, y: y))
                    queue.append(CGPoint.init(x: x-1, y: y))
                    queue.append(CGPoint.init(x: x, y: y+1))
                    queue.append(CGPoint.init(x: x, y: y-1))
                    visited[x][y] = true;
                }
            }
            queue.removeFirst()
        }
    }
    
    func DFSWithStack(_ grid : [[Character]], i : Int, j : Int) {
        

        if(i>=0 && i<rows && j>=0 && j<cols ){
            stack.append(CGPoint.init(x: i, y: j))
            let x = Int(stack.last!.x)
            let y = Int(stack.last!.y)
            if(grid[x][y] == "0"  || visited[x][y] == true ){
            stack.removeLast()
            return;
            }
            visited[x][y] = true;
            DFSWithStack(grid, i: x+1, j: y)
            DFSWithStack(grid, i: x-1, j: y)
            DFSWithStack(grid, i: x, j: y+1)
            DFSWithStack(grid, i: x, j: y-1)
        }

    }
    

}

swift不熟,不够优雅代码有冗余,轻喷!

©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

友情链接更多精彩内容