leetcode：139. 单词拆分（字典树法+剪枝）3ms

最新推荐文章于 2022-09-21 11:49:33 发布

原创最新推荐文章于 2022-09-21 11:49:33 发布 · 521 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#java #字典树 #剪枝

算法试题（java）专栏收录该内容

177 篇文章

订阅专栏

本文介绍了一种利用字典树（Trie）的数据结构来高效解决单词拆分问题的方法。通过将字典中的单词插入到字典树中，可以快速判断一个给定的字符串是否能被拆分为字典中出现的一个或多个单词。文章详细阐述了算法的实现过程，包括单词插入、查找以及如何使用布尔数组记录拆分状态。

给定一个非空字符串 s 和一个包含非空单词列表的字典 wordDict，判定 s 是否可以被空格拆分为一个或多个在字典中出现的单词。

说明：

拆分时可以重复使用字典中的单词。
你可以假设字典中没有重复的单词。

示例 1：

输入: s = "leetcode", wordDict = ["leet", "code"]
输出: true
解释: 返回 true 因为 "leetcode" 可以被拆分成 "leet code"。

示例 2：

输入: s = "applepenapple", wordDict = ["apple", "pen"]
输出: true
解释: 返回 true 因为 "applepenapple" 可以被拆分成 "apple pen apple"。
     注意你可以重复使用字典中的单词。

示例 3：

输入: s = "catsandog", wordDict = ["cats", "dog", "sand", "and", "cat"]
输出: false

package LeetCode;

import java.util.ArrayList;
import java.util.List;

public class WordBreak139 {
    TrieNode root;

    public WordBreak139() {
        root = new TrieNode();
    }
    //把单词加入到字典树中
    public void insert(String w) {
        TrieNode r = root;
        for (int i = 0; i < w.length(); i++) {
            int index = w.charAt(i) - 'a';
            if (r.childs[index] == null) {
                r.childs[index] = new TrieNode();
            }
            r = r.childs[index];
        }
        r.isWord = true;
    }

    public boolean wordBreak(String s, List<String> wordDict) {
        //把每个lise中单词加入到字典树中
        for (int i = 0; i < wordDict.size(); i++) {
            String ws = wordDict.get(i);
            insert(ws);
        }
        //用来记录上个单词结束（或者说记录下一个字典中词的开始位置）第一个位置是入口所以设为true
        boolean[] juge = new boolean[s.length() + 1];
        juge[0] = true;
        //
        for (int i = 0; i < s.length(); i++) {
            //要在记录的字典树中所包含了的后面一个开始也就是前一个字典已找出没必要在所属字中找另一个字
            if (juge[i] == true) {
                //每次搜索初始化字典开始
                TrieNode r = root;
                //字典树的搜索
                for (int j = i; j < s.length(); j++) {
                     int index =s.charAt(j) - 'a';
                     //判断是否字符在字典树中（判断字符是否为字典中的字或者是字典中的前缀）
                    //如果不是前缀或者不是字典中的词则直接结束循环
                    if (r != null && r.childs[index] != null) {
                        r = r.childs[index];
                        if (r.isWord) {
                            juge[j + 1] = true;
                        }
                    } else {
                        break;
                    }
                }
            }
        }
        return juge[s.length()];
    }

    public static void main(String[] args) {
        WordBreak139 w = new WordBreak139();
        List<String> strings = new ArrayList<>();
        strings.add("leet");
        strings.add("code");
        w.wordBreak("leetcode", strings);
    }
}