leetcode 212. Word Search II 单词矩阵搜索 + DFS + 字典树

Given a 2D board and a list of words from the dictionary, find all words in the board.

Each word must be constructed from letters of sequentially adjacent cell, where “adjacent” cells are those horizontally or vertically neighboring. The same letter cell may not be used more than once in a word.

For example,
Given words = [“oath”,”pea”,”eat”,”rain”] and board =

[
[‘o’,’a’,’a’,’n’],
[‘e’,’t’,’a’,’e’],
[‘i’,’h’,’k’,’r’],
[‘i’,’f’,’l’,’v’]
]
Return [“eat”,”oath”].
Note:
You may assume that all inputs are consist of lowercase letters a-z.

这道题最直觉的方法就是DFS深度优先搜索,但是会发现超时,这个我也么办法了,这个方法是最直接的方法,我认为是最好的方法。

我在网上找到了一个使用字典树的应用,说实话这个我想不到,但是还是这么做时间空间效率最好。

代码如下:

import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;

/*
 * 典型的DFS深度优先搜索
 * 不过会超时
 * */
class Solution 
{
    public List<String> findWords(char[][] board, String[] words)
    {
        //这里res使用set是为了避免添加重复的元素
        Set<String> res=new HashSet<String>();
        if(board==null || words==null || board.length<=0 || words.length<=0)
            return new ArrayList<>(res);

        List<String> target=new ArrayList<>();
        for(int i=0;i<words.length;i++)
            target.add(words[i]);

        boolean [][]visit=new boolean [board.length][board[0].length];

/*      StringBuilder builder=new StringBuilder();
        for(int i=0;i<board.length;i++)
        {
            for(int j=0;j<board[0].length;j++)
            {
                search(board,visit,builder,target,res,i,j);
            }
        }*/


        /*
         * 使用字典树剪枝,加速搜索
         * */
        Trie trie = new Trie();
        for(String word : words)
            trie.insert(word);
        for(int i=0; i<board.length; i++) 
            for(int j=0; j<board[0].length; j++) 
                search(board, visit, trie, i, j, new StringBuilder(), res);

        return new ArrayList<>(res);
    }

    private void search(char[][] board,boolean[][] visited,Trie trie,int i,int j, StringBuilder sb, Set<String> res)
    {
        if(i<0 || i>board.length-1 || j<0 || j>board[0].length-1 || visited[i][j])  return;
        sb.append(board[i][j]);
        String s = sb.toString();
        visited[i][j] = true;
        if(trie.startsWith(s)) 
        {
            if(trie.search(s)) res.add(s);

            search(board, visited, trie, i-1, j, sb, res);
            search(board, visited, trie, i+1, j, sb, res);
            search(board, visited, trie, i, j-1, sb, res);
            search(board, visited, trie, i, j+1, sb, res);
        }
        sb.deleteCharAt(sb.length() - 1);
        visited[i][j] = false;
    }


    /*
     * 使用DFS直接搜索
     * */
    private void search(char[][] board, boolean[][] visit,
            StringBuilder builder, List<String> target, Set<String> res,
            int i, int j) 
    {
        if(i>=0&&i<board.length && j>=0&&j<board[0].length && visit[i][j]==false)
        {
            builder.append(board[i][j]);
            visit[i][j]=true;
            if(target.contains(builder.toString()))
                res.add(builder.toString());

            search(board, visit, builder, target, res, i-1, j);
            search(board, visit, builder, target, res, i+1, j);
            search(board, visit, builder, target, res, i, j+1);
            search(board, visit, builder, target, res, i, j-1);

            visit[i][j]=false;
            builder.deleteCharAt(builder.length()-1);
        }else 
            return ;

    }
}


class TrieNode 
{
    // Initialize your data structure here.
    char c;
    boolean leaf;
    HashMap<Character, TrieNode> children = new HashMap<Character, TrieNode>();
    public TrieNode(char c) {
        this.c = c;
    }
    public TrieNode(){};
}

class Trie 
{
    private TrieNode root;
    public Trie() {
        root = new TrieNode();
    }

    // Inserts a word into the trie.
    public void insert(String word) 
    {
        Map<Character, TrieNode> children = root.children;
        for(int i=0; i<word.length(); i++)
        {
            char c = word.charAt(i);
            TrieNode t;
            if(children.containsKey(c))
                t = children.get(c);
            else 
            {
                t = new TrieNode(c);
                children.put(c, t);
            }
            children = t.children;
            if(i==word.length()-1) t.leaf=true;
        }
    }

    // Returns if the word is in the trie.
    public boolean search(String word)
    {
        TrieNode t = searchNode(word);
        return t!=null && t.leaf;
    }

    public boolean startsWith(String prefix)
    {
        return searchNode(prefix) != null;
    }

    private TrieNode searchNode(String word) 
    {
        Map<Character, TrieNode> children = root.children;
        TrieNode t = null;
        for(int i=0; i<word.length(); i++)
        {
            char c = word.charAt(i);
            if(!children.containsKey(c)) 
                return null;
            t = children.get(c);
            children = t.children;
        }
        return t;
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值