Swift | LeetCode

Задача: 79. Word Search
Сложность: medium

Дана сетка символов размером m на n, называемая board, и строка word. Верните true, если слово word существует в сетке.

Слово можно составить из букв последовательно смежных ячеек, где смежные ячейки находятся рядом по горизонтали или вертикали. Одна и та же ячейка с буквой не может быть использована более одного раза.

Пример:

Input: board = [["A","B","C","E"],["S","F","C","S"],["A","D","E","E"]], word = "ABCCED"
Output: true

👨‍💻

Алгоритм:

1⃣Общий подход к алгоритмам обратной трассировки: В каждом алгоритме обратной трассировки существует определенный шаблон кода. Например, один из таких шаблонов можно найти в нашем разделе "Рекурсия II". Скелет алгоритма представляет собой цикл, который проходит через каждую ячейку в сетке. Для каждой ячейки вызывается функция обратной трассировки (backtrack()), чтобы проверить, можно ли найти решение, начиная с этой ячейки.

2⃣Функция обратной трассировки: Эта функция, реализуемая как алгоритм поиска в глубину (DFS), часто представляет собой рекурсивную функцию. Первым делом проверяется, достигнут ли базовый случай рекурсии, когда слово для сопоставления пусто, то есть для каждого префикса слова уже найдено совпадение. Затем проверяется, не является ли текущее состояние недопустимым: либо позиция ячейки выходит за границы доски, либо буква в текущей ячейке не совпадает с первой буквой слова.

3⃣Исследование и завершение: Если текущий шаг допустим, начинается исследование с использованием стратегии DFS. Сначала текущая ячейка помечается как посещенная, например, любой неалфавитный символ подойдет. Затем осуществляется итерация через четыре возможных направления: вверх, вправо, вниз и влево. Порядок направлений может быть изменен по предпочтениям пользователя. В конце исследования ячейка возвращается к своему исходному состоянию, и возвращается результат исследования.

😎

Решение:

func exist(_ board: inout [[Character]], _ word: String) -> Bool {
    let rows = board.count
    let cols = board[0].count

    func backtrack(_ row: Int, _ col: Int, _ index: Int) -> Bool {
        if index == word.count { return true }
        if row < 0 || row >= rows || col < 0 || col >= cols || board[row][col] != word[word.index(word.startIndex, offsetBy: index)] {
            return false
        }
        
        board[row][col] = "#"  
        let directions = [(0, 1), (1, 0), (0, -1), (-1, 0)]
        var result = false
        
        for (rowOffset, colOffset) in directions {
            result = backtrack(row + rowOffset, col + colOffset, index + 1)
            if result { break }
        }
        
        board[row][col] = word[word.index(word.startIndex, offsetBy: index)]  
        return result
    }

    for row in 0..<rows {
        for col in 0..<cols {
            if backtrack(row, col, 0) { return true }
        }
    }
    return false
}

Ставь 👍 и забирай 📚 Базу знаний

Please open Telegram to view this post