212. 单词搜索 II

给定一个二维网格 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数组,匹配到字典树的字符进行深度优先遍历

class Solution {
    private char[][] board;
    private TrieNdoe root = new TrieNdoe();
    private Set<String> result = new HashSet<>();
    private int[][] dir = { // 上下左右
            {0, 1}, {0, -1}, {1, 0}, {-1, 0}
    };
    private boolean[][] used;

    public List<String> findWords(char[][] board, String[] words) {
        this.board = board;
        this.used = new boolean[board.length][board[0].length];
        for (String word : words) {
            insert(word);
        }
        for (int i = 0; i < board.length; i++) {
            for (int j = 0; j < board[i].length; j++) {
                findWord(i, j, root);
            }
        }
        return new ArrayList<>(result);
    }

    /**
     * DFS深度优先搜索
     *
     * @param row
     * @param column
     * @param node
     */
    private void findWord(int row, int column, TrieNdoe node) {
        char ch = board[row][column];
        int chIndex = ch - 'a';
        // 没找到 或者 已经使用过了
        if (node.nextNode[chIndex] == null || used[row][column]) {
            return;
        }
        // 如果找到一个word就记录下来
        if (node.nextNode[chIndex].endOfWord) {
            result.add(node.nextNode[chIndex].word);
        }
        // 记录当前位置已经使用过了
        used[row][column] = true;
        // 上下左右深度搜索
        for (int[] d : dir) {
            if (row + d[0] < 0 || row + d[0] >= board.length || column + d[1] < 0 || column + d[1] >= board[0].length) {
                continue;
            }
            findWord(row + d[0], column + d[1], node.nextNode[chIndex]);
        }
        // 回溯,将当前位置置为未使用过
        used[row][column] = false;
    }

    /**
     * 构建字典树
     */
    private void insert(String word) {
        TrieNdoe node = root;
        int chIndex;
        for (int i = 0; i < word.length(); i++) {
            chIndex = word.charAt(i) - 'a';
            if (node.nextNode[chIndex] == null) {
                node.nextNode[chIndex] = new TrieNdoe();
            }
            node = node.nextNode[chIndex];
        }
        node.endOfWord = true;
        node.word = word;
    }

    private class TrieNdoe {
        // 假设所有输入都由小写字母 a-z 组成
        public TrieNdoe[] nextNode = new TrieNdoe['z' - 'a' + 1];
        // 如果是最后的节点,才存
        public String word; 
        public boolean endOfWord;
    }

    public static void main(String[] args) {
        char[][] board = {{'a', 'b'}};
        new Solution().findWords(board, new String[]{"ba"});
    }
}
运行效率
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 159,015评论 4 362
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 67,262评论 1 292
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 108,727评论 0 243
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 43,986评论 0 205
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,363评论 3 287
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,610评论 1 219
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,871评论 2 312
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,582评论 0 198
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,297评论 1 242
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,551评论 2 246
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 32,053评论 1 260
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,385评论 2 253
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 33,035评论 3 236
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,079评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,841评论 0 195
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,648评论 2 274
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,550评论 2 270

推荐阅读更多精彩内容