JAVA程序设计：单词搜索 II（LeetCode：212）

给定一个二维网格 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（前缀树）。

 

思路：字典树+爆搜即可，没有什么难度和坑点，记得用set存一下，因为不用set你需要花费额外的时间去判断答案中是否有重复单词。

class Solution {
	class Node
	{
		Node[] next;
		boolean isWord;
		public Node()
		{
			next=new Node[26];
			isWord=false;
		}
	}
	
    Node root;
	Set<String> ans=new HashSet<>();
	
    public List<String> findWords(char[][] board, String[] words) {
    	root=new Node();
        for(String str:words)
        {
        	Node cur=root;
        	for(int i=0;i<str.length();i++)
        	{
        		int c=str.charAt(i)-'a';
        		if(cur.next[c]==null)
        			cur.next[c]=new Node();
        		cur=cur.next[c];
        	}
        	cur.isWord=true;
        }
        
        int m=board.length;
        int n=board[0].length;
        boolean[][] visited=new boolean[m][n];
        
        for(int i=0;i<m;i++)
        	for(int j=0;j<n;j++)
        	{
        		if(root.next[board[i][j]-'a']!=null)
        			dfs(board,i,j,m,n,root.next[board[i][j]-'a'],board[i][j]+"",visited);
        	}
        
        return new LinkedList<String>(ans);
    }
    
    private void dfs(char[][] board,int x,int y,int m,int n,Node root,String str,boolean[][] visited)
    {
    	if(x>=m || y>=n || x<0 || y<0 || visited[x][y])
    		return;
    	
    	visited[x][y]=true;
    	
    	if(root.isWord)
    		ans.add(str);
    	
    	if(y+1<n && root.next[board[x][y+1]-'a']!=null)
    		dfs(board,x,y+1,m,n,root.next[board[x][y+1]-'a'],str+board[x][y+1],visited);
    	if(x+1<m && root.next[board[x+1][y]-'a']!=null)
    		dfs(board,x+1,y,m,n,root.next[board[x+1][y]-'a'],str+board[x+1][y],visited);
    	if(y-1>=0 && root.next[board[x][y-1]-'a']!=null)
    		dfs(board,x,y-1,m,n,root.next[board[x][y-1]-'a'],str+board[x][y-1],visited);
    	if(x-1>=0 && root.next[board[x-1][y]-'a']!=null)
    		dfs(board,x-1,y,m,n,root.next[board[x-1][y]-'a'],str+board[x-1][y],visited);
    	
    	visited[x][y]=false;
    }
}