糖逗 发表于 2020-12-19 16:38:54

C++刷LeetCode(720. 词典中最长的单词)【字典树】【深度优先搜索】

题目描述:
给出一个字符串数组words组成的一本英语词典。从中找出最长的一个单词,该单词是由words词典中其他单词逐步添加一个字母组成。若其中有多个可行的答案,则返回答案中字典序最小的单词。

若无答案,则返回空字符串。

 

示例 1:

输入:
words = ["w","wo","wor","worl", "world"]
输出:"world"
解释:
单词"world"可由"w", "wo", "wor", 和 "worl"添加一个字母组成。
示例 2:

输入:
words = ["a", "banana", "app", "appl", "ap", "apply", "apple"]
输出:"apple"
解释:
"apply"和"apple"都能由词典中的单词组成。但是"apple"的字典序小于"apply"。
 

提示:

所有输入的字符串都只包含小写字母。
words数组长度范围为。
words的长度范围为。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/longest-word-in-dictionary
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。


class Solution {
public:
    struct TrieTree{
      bool flag;
      map<char, TrieTree*>next;
      TrieTree():flag(false){};
    };
    void dfs(TrieTree* cur_node, string&res, string temp){
      if(cur_node == NULL ||cur_node -> flag == false)return;
      if(temp.size() > res.size())res = temp;
      for(auto cha : cur_node -> next){
            dfs(cur_node -> next, res, temp + cha.first);
      }
    }
    string longestWord(vector<string>& words) {
      //构建字典树
      TrieTree* root = new TrieTree();
      for(auto word : words){
            TrieTree* node = root;
            for(auto cha : word){
                if((node -> next).count(cha) == 0){
                  node -> next = new TrieTree();
                }
                node = node -> next;
            }
            node -> flag = true;
      }
      //深度优先搜索
      string res;
      for(auto cha : root ->next){
            if(root -> next -> flag == true){
                string temp;
                dfs(root -> next, res, temp + cha.first);
            }
      }
      return res;
    }
};

糖逗 发表于 2020-12-19 16:39:24

{:10_284:}
页: [1]
查看完整版本: C++刷LeetCode(720. 词典中最长的单词)【字典树】【深度优先搜索】