LeetCode - #212 单词搜索 II

最新推荐文章于 2025-08-13 18:53:41 发布

网罗开发

最新推荐文章于 2025-08-13 18:53:41 发布

阅读量677

点赞数 15

CC 4.0 BY-SA版权

分类专栏： Swift 文章标签： vue.js leetcode 算法

本文链接：https://blog.youkuaiyun.com/qq_36478920/article/details/145787798

Swift 专栏收录该内容

348 篇文章

订阅专栏

在这里插入图片描述

网罗开发 （小红书、快手、视频号同名）

大家好，我是展菲，目前在上市企业从事人工智能项目研发管理工作，平时热衷于分享各种编程领域的软硬技能知识以及前沿技术，包括iOS、前端、Harmony OS、Java、Python等方向。在移动端开发、鸿蒙开发、物联网、嵌入式、云原生、开源等领域有深厚造诣。

图书作者：《ESP32-C3 物联网工程开发实战》
图书作者：《SwiftUI 入门，进阶与实战》
超级个体：COC上海社区主理人
特约讲师：大学讲师，谷歌亚马逊分享嘉宾
科技博主：极星会首批签约作者

文章目录

前言

我们社区陆续会将顾毅（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]()
        }
    }
}