JAVA程序设计：单词替换（LeetCode：648）

在英语中，我们有一个叫做 词根(root)的概念，它可以跟着其他一些词组成另一个较长的单词——我们称这个词为 继承词(successor)。例如，词根an，跟随着单词 other(其他)，可以形成新的单词 another(另一个)。

现在，给定一个由许多词根组成的词典和一个句子。你需要将句子中的所有继承词用词根替换掉。如果继承词有许多可以形成它的词根，则用最短的词根替换它。

你需要输出替换之后的句子。

示例 1:

输入: dict(词典) = ["cat", "bat", "rat"]
sentence(句子) = "the cattle was rattled by the battery"
输出: "the cat was rat by the bat"
注:

输入只包含小写字母。
1 <= 字典单词数 <=1000
1 <=  句中词语数 <= 1000
1 <= 词根长度 <= 100
1 <= 句中词语长度 <= 1000

方法一：因为单词数以及语句中出现的单词不超过1000，我们可以采用暴力方法遍历每个串的前缀看是否出现过，如果出现过直接break即可。

class Solution {
    public String replaceWords(List<String> dict, String sentence) {
        
    	int len=0;
    	String[] s=new String[sentence.length()];
    	StringBuilder tmp=new StringBuilder();
    	StringBuilder str=new StringBuilder();
        
    	for(int i=0;i<sentence.length();i++) {
    		if(sentence.charAt(i)==' ' || i==sentence.length()-1) {
    			if(i==sentence.length()-1) 
    				tmp.append(sentence.charAt(i));
    			s[len++]=tmp.toString();
    			tmp.delete(0, tmp.length());
    			continue;
    		}
    		tmp.append(sentence.charAt(i));
    	}
    	
    	for(int i=0;i<len;i++) {
    		boolean flag=false;
    		for(int j=0;j<s[i].length();j++) {	
    			if(dict.contains(s[i].substring(0, j+1))) {
    				flag=true;
    				str.append(s[i].substring(0, j+1));
    				break;
    			} 
    		}
    		if(!flag) str.append(s[i]);
    		if(i+1<len) str.append(" ");
    	}
    	
    	return str.toString();
    }
}

方法二：把所有的词根放入前缀树中，在树上查找每个单词的最短词根，该操作可在线性时间内完成。

class Solution {
	class TrieNode{
		
		String word;
		TrieNode[] children;
		
		public TrieNode() {
			word=null;
			children=new TrieNode[26];
		}
	}
	
    public String replaceWords(List<String> dict, String sentence) {
        
    	TrieNode trie=new TrieNode();
    	for(String root : dict) {
    		TrieNode cur=trie;
    		for(char letter : root.toCharArray()) {
    			if(cur.children[letter-'a']==null)
    				cur.children[letter-'a']=new TrieNode();
    			cur=cur.children[letter-'a'];	
    		}
    		cur.word=root;
    	}
    	
    	StringBuilder ans=new StringBuilder();
    	
    	for(String word : sentence.split("\\s+")) {
    		if(ans.length()>0) ans.append(" ");
    		
    		TrieNode cur=trie;
    		for(char letter : word.toCharArray()) {
    			if(cur.children[letter-'a']==null || cur.word!=null)
    				break;
    			cur=cur.children[letter-'a'];
    		}
    		ans.append(cur.word!=null?cur.word:word);
    	}
    	
    	return ans.toString();
    }
}