【java|golang】648. 单词替换---字典树

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

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

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

示例 1：

输入：dictionary = [“cat”,“bat”,“rat”], sentence = “the cattle was rattled by the battery”
输出：“the cat was rat by the bat”
示例 2：

输入：dictionary = [“a”,“b”,“c”], sentence = “aadsfasf absbs bbab cadsfafs”
输出：“a a b c”

提示：

1 <= dictionary.length <= 1000
1 <= dictionary[i].length <= 100
dictionary[i] 仅由小写字母组成。
1 <= sentence.length <= 10^6
sentence 仅由小写字母和空格组成。
sentence 中单词的总量在范围 [1, 1000] 内。
sentence 中每个单词的长度在范围 [1, 1000] 内。
sentence 中单词之间由一个空格隔开。
sentence 没有前导或尾随空格。

public String replaceWords(List<String> dictionary, String sentence) {
        dictionary.sort(Comparator.comparing(String::length));
        StringBuilder sb = new StringBuilder();
        dp:
        for (String s : sentence.split(" ")) {
            for (String dic : dictionary) {
                 if(s.startsWith(dic)){
                     sb.append(dic);
                     sb.append(" ");
                     continue dp;
                 }
            }
            sb.append(s);
            sb.append(" ");
        }
        return sb.toString().trim();
    }

class TrieNode {
        boolean isLeaf;
        TrieNode[] children = new TrieNode[26];
    }

    TrieNode root = new TrieNode();

    public String replaceWords(List<String> dict, String sentence) {
        add(dict);
        StringBuilder stringBuilder = new StringBuilder();
        for(String s : sentence.split(" ")){
            int index = search(s);
            String str = index > 0 ? s.substring(0, index) : s;
            stringBuilder.append(str);
            stringBuilder.append(" ");
        }
        return stringBuilder.toString().trim();
    }

    private int search(String s) {
        TrieNode node = root;
        int index = 0;
        for(char c : s.toCharArray()){
            if(node.children[c-'a'] == null && !node.isLeaf) return 0;
            if(node.isLeaf) return index;
            index++;
            node = node.children[c-'a'];
        }
        return index;

    }

    private void add(List<String> dict) {
        for(String d : dict){
            TrieNode node = root;
            for (char c : d.toCharArray()) {
                if (node.children[c - 'a'] == null) {
                    node.children[c - 'a'] = new TrieNode();
                }
                node = node.children[c - 'a'];
            }
            node.isLeaf = true;
        }
    }

func replaceWords(dictionary []string, sentence string) string {
   sort.Slice(dictionary, func(i, j int) bool {
	   return len(dictionary[j])>len(dictionary[i])
   })
   split := strings.Split(sentence, " ")
    s := make([]string,0)
	dp:for _, v := range split {
		for _, dic := range dictionary {
			if strings.HasPrefix(v,dic){
				s = append(s, dic)
				continue dp
			}
		}
		s = append(s, v)
	}
    return strings.Join(s," ")
}