leetcode 720. 词典中最长的单词(Longest Word in Dictionary)

目录

• 题目描述：
• 示例 1:
• 示例 2:
• 解法：

题目描述：

给出一个字符串数组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数组长度范围为[1,1000]。
• words[i]的长度范围为[1,30]。

---

解法：

# define PR pair<int, string>

class Solution {
public:
    string longestWord(vector<string>& words) {
        string res = "";
        int len = 0;
        unordered_set<string> st;
        st.insert("");
        vector<PR> lst;
        for(string word : words){
            lst.push_back({word.size(), word});
        }
        sort(lst.begin(), lst.end());
        for(PR pr : lst){
            string word = pr.second;
            int sz = pr.first;
            if(st.find(word.substr(0, sz-1)) != st.end()){
                if(len < sz){
                    len = sz;
                    res = word;
                }
                st.insert(word);
            }
        }
        return res;
    }
};