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数组长度范围为[1,1000]。
• words[i]的长度范围为[1,30]。

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

执行结果

代码描述

思路：1. 将字符串数组由大到小进行排序，特殊处理一下第一个。

2. 判断更小的字符串是否是当前字符串的子集，这种子集仅差1个字符。

3. 是的话，就存下来，不是的话，就继续扫描后面的。

class Solution {
public:
    string longestWord(vector<string>& words) {
        sort(words.begin(), words.end());
        set<string> str;
        string res = "";
        for(int i = 0; i < words.size(); ++i)
        {
            if(words[i].size() == 1 || str.count(words[i].substr(0, words[i].size() - 1)) == 1)
            {
                res = words[i].size() > res.size() ? words[i] : res;
                str.insert(words[i]);
            }
        }
        return res;
    }
};