LeetCode - #212 单词搜索 II

持续创作，加速成长！这是我参与「掘金日新计划 · 10 月更文挑战」的第16天，点击查看活动详情

前言

我们社区陆续会将顾毅（Netflix 增长黑客，《iOS 面试之道》作者，ACE 职业健身教练。）的 Swift 算法题题解整理为文字版以方便大家学习与阅读。

LeetCode 算法到目前我们已经更新到 211 期，我们会保持更新时间和进度（周一、周三、周五早上 9:00 发布），每期的内容不多，我们希望大家可以在上班路上阅读，长久积累会有很大提升。

不积跬步，无以至千里；不积小流，无以成江海，Swift社区 伴你前行。如果大家有建议和意见欢迎在文末留言，我们会尽力满足大家的需求。

难度水平：困难

1. 描述

给定一个 m x n 二维字符网格 board 和一个单词（字符串）列表 words， 返回所有二维网格上的单词 。

单词必须按照字母顺序，通过 相邻的单元格 内的字母构成，其中“相邻”单元格是那些水平相邻或垂直相邻的单元格。同一个单元格内的字母在一个单词中不允许被重复使用。

2. 示例

示例 1

输入：board = [["o","a","a","n"],["e","t","a","e"],["i","h","k","r"],["i","f","l","v"]], words = ["oath","pea","eat","rain"]
输出：["eat","oath"]

示例 2

输入：board = [["a","b"],["c","d"]], words = ["abcb"]
输出：[]

提示：

• m == board.length
• n == board[i].length
• 1 <= m, n <= 12
• board[i][j] 是一个小写英文字母
• 1 <= words.length <= 3 * 10^4
• 1 <= words[i].length <= 10
• words[i] 由小写英文字母组成
• words 中的所有字符串互不相同

3. 答案

class WordSearchII {
    
    func findWords(_ board: [[Character]], _ words: [String]) -> [String] {
        let trie = Trie(words), m = board.count, n = board[0].count
        var isVisited = Array(repeating: Array(repeating: false, count: n), count: m), res = Set<String>()
        
        for i in 0..<m {
            for j in 0..<n {
                search(board, trie, &res, i, j, &isVisited, trie.root, "", m, n)
            }
        }
        
        return Array(res)
    }
    
    private func search(_ board: [[Character]], _ trie: Trie, _ res: inout Set<String>, _ i: Int, _ j: Int, _ isVisited: inout [[Bool]], _ currentNode: TrieNode, _ currentStr: String, _ m: Int, _ n: Int) {
        guard i >= 0 && i < m && j >= 0 && j < n else {
            return
        }
        
        guard !isVisited[i][j] else {
            return
        }
        
        guard let child = currentNode.children[board[i][j]] else {
            return
        }
        
        isVisited[i][j] = true
            
        let str = currentStr + "\(board[i][j])"
            
        if child.isEnd {
            res.insert(str)
        }
        
        search(board, trie, &res, i + 1, j, &isVisited, child, str, m, n)
        search(board, trie, &res, i - 1, j, &isVisited, child, str, m, n)
        search(board, trie, &res, i, j + 1, &isVisited, child, str, m, n)
        search(board, trie, &res, i, j - 1, &isVisited, child, str, m, n)
            
        isVisited[i][j] = false
    }
    
    class Trie {
        var root: TrieNode
        
        init(_ words: [String]) {
            root = TrieNode()
            
            words.forEach { insert($0) }
        }
        
        private func insert(_ word: String) {
            var node = root
            
            for char in word {
                if node.children[char] == nil {
                    node.children[char] = TrieNode()
                }
                
                node = node.children[char]!
            }
            
            node.isEnd = true
        }
    }
    
    class TrieNode {
        var isEnd: Bool
        var children: [Character: TrieNode]
        
        init() {
            isEnd = false
            children = [Character: TrieNode]()
        }
    }
}

• 主要思想：经典的深度优先搜索，在 Trie 的帮助下上，下，左，右四个方向。
• 时间复杂度：O(mn^2) m 是单词中一个单词最长的长度
• 空间复杂度：O(n^2)

该算法题解的仓库：LeetCode-Swift

点击前往 LeetCode 练习

关于我们

我们是由 Swift 爱好者共同维护，我们会分享以 Swift 实战、SwiftUI、Swift 基础为核心的技术内容，也整理收集优秀的学习资料。