leetcode 212. 单词搜索 II Java 实现

关键词

回溯 trie(前缀树)

题目描述

给定一个二维网格 board 和一个字典中的单词列表 words，找出所有同时在二维网格和字典中出现的单词。

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

示例:

输入:

words = ["oath","pea","eat","rain"] and board =
[
  ['o','a','a','n'],
  ['e','t','a','e'],
  ['i','h','k','r'],
  ['i','f','l','v']
]

输出: ["eat","oath"]
说明:
你可以假设所有输入都由小写字母 a-z 组成。

提示：你需要优化回溯算法以通过更大数据量的测试。你能否早点停止回溯？
如果当前单词不存在于所有单词的前缀中，则可以立即停止回溯。什么样的数据结构可以有效地执行这样的操作？散列表是否可行？为什么？ 前缀树如何？如果你想学习如何实现一个基本的前缀树，请先查看这个问题： 实现Trie（前缀树）。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/word-search-ii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解答

低端解法

遍历 board，尝试以每个位置的单词为开头，向四周递归遍历，匹配目标单词。

• 记录遍历过程中的 trie 树的结点状态
• 回溯后的函数变量状态重置

import java.util.*;

class Solution {
   
    class TrieNode {
   
        /** 26 English character */
        public TrieNode[] childs = new TrieNode[26];
        public boolean isEnd = false;
        public char value;
    }
    
    class Trie {
   

        TrieNode root;

        // 用于单字符查找回溯的栈和变量
        Stack<TrieNode> stack = new Stack<>();
        TrieNode cNode;

        /** Initialize your data structure here. */
        public Trie() {
   
            root = new TrieNode();
            cNode = root;
        }

        /** Inserts a word into the trie. */
        public void insert(String word) {
   
            TrieNode node = root;
            for(int i=0;i<word.length();i++) {
   
                int chIndex = word.charAt(i)-'a'; 
                if(node.childs[chIndex] == null) {
   
                    node.childs[chIndex] = new TrieNode();
                    node.childs[chIndex].value = word.charAt(i);