720. 词典中最长的单词

83 阅读1分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第31天,点击查看活动详情

一、题目描述:

720. 词典中最长的单词 - 力扣(LeetCode) (leetcode-cn.com)

给出一个字符串数组 words 组成的一本英语词典。返回 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" 

提示:

  • 1 <= words.length <= 1000
  • 1 <= words[i].length <= 30
  • 所有输入的字符串 words[i] 都只包含小写字母。

二、思路分析:

  1. 建一个字典树,遍历word加入字典树。但是结果需要在遍历一次words[i], 判断每个words[i]减少一个字母是否在字典树前缀中,因为可能先出现ap,后出现a,建字典树的遍历不知道ap也是符合要求的
  2. 所以先用qsort对words按strlen结果排个序。

三、AC 代码:

typedef struct dictTree {
    int c;
    int isEnd;
    struct dictTree *next[26];
} DICTTREE;

int insertWord(DICTTREE *d, char *s) {
    int i;
    bool flag = true;
    DICTTREE *cur = d;
    for (i=0; i<strlen(s); i++) {
        if (cur->next[s[i]-'a'] == NULL) {
            cur->next[s[i]-'a'] = (DICTTREE*)malloc(sizeof(DICTTREE));
            (void)memset(cur->next[s[i]-'a'], 0, sizeof(DICTTREE));
            cur->next[s[i]-'a']->c = s[i];
        }
        if (i == strlen(s)-1) {
            cur->next[s[i]-'a']->isEnd = true;
            return flag;
        }
        if (cur->next[s[i]-'a']->isEnd == false) {
            flag = false;
        }
        cur = cur->next[s[i]-'a'];
    }
    return flag;
}

int cmp(const void *a, const void *b) {
    return strlen(*(char**)a) - strlen(*(char**)b);
}

char * longestWord(char ** words, int wordsSize){
    qsort(words, wordsSize, sizeof(char*), cmp);
    DICTTREE d = {0};
    char *ans = NULL;
    int i;
    for (i=0; i<wordsSize; i++) {
        if (!insertWord(&d, words[i])) continue;

        //printf("complete word %s\n", words[i]);

        if (!ans) ans = words[i];
        else if (strlen(words[i]) > strlen(ans) ||
            (strlen(words[i]) == strlen(ans) && strcmp(words[i], ans) < 0)) {       
            ans = words[i];
        }
    }

    return ans;
}

范文参考

字典树的应用-查找最长单词-qsort - 词典中最长的单词 - 力扣(LeetCode)