leetcode 820.单词的压缩码 笔记

题目链接
给定一个单词列表,我们将这个列表编码成一个索引字符串 S 与一个索引列表 A。
例如,如果这个列表是 [“time”, “me”, “bell”],我们就可以将其表示为 S = “time#bell#” 和 indexes = [0, 2, 5]。
对于每一个索引,我们可以通过从字符串 S 中索引的位置开始读取字符串,直到 “#” 结束,来恢复我们之前的单词列表。
那么成功对给定单词列表进行编码的最小字符串长度是多少呢?

示例:

输入: words = ["time", "me", "bell"]
输出: 10
说明: S = "time#bell#" , indexes = [0, 2, 5]

题解方法用的是讲解里的字典法。
字典法本身比较好理解,将每个单词倒着把字母一次存入树中。图片来自leetcode解析
但在看完代码最后的for循环时懵了——

        for (auto& [node, idx] : nodes) {
            if (node->count == 0) {
                ans += words[idx].length() + 1;
            }
        }

这循环咋个用的,哈希表存的不是字母结尾或者开头吗??

看了会儿才想明白,它的哈希表nodes里存的键值对,键是每个单词开头字母节点的指针,且第一次创建该节点时对应的count=0。而值并不是字母表,而是每个单词在vector里的位置
上图中的粉色节点才是存在哈希表nodes里的东西
到这里才是真的搞清楚了做法,但感觉自己去想依旧很难想象,还是得多做题。

代码如下:

class TrieNode {
    TrieNode* children[26];
public:
    int count;
    TrieNode() {
        for (int i = 0; i < 26; ++i) children[i] = NULL;
        count = 0;
    }
    TrieNode* get(char c) {
        if (children[c - 'a'] == NULL) {
            children[c - 'a'] = new TrieNode();
            count++;
        }
        return children[c - 'a'];
    }
};
class Solution {
public:
    int minimumLengthEncoding(vector<string>& words) {
        TrieNode* trie = new TrieNode();
        unordered_map<TrieNode*, int>nodes;
        int si = words.size();
        for (int i = 0; i < si; i++) {
            string word = words[i];
            TrieNode* cur = trie;
            for (int j = word.length() - 1; j >= 0; --j)
                cur = cur->get(word[j]);
            nodes[cur] = i;
        }
        int ans = 0;
        for (auto iter : nodes) {
            if (iter.first->count == 0) {
                ans += words[iter.second].length()+1;
            }
        }
        return ans;
    }
};
发布了9 篇原创文章 · 获赞 1 · 访问量 150

猜你喜欢

转载自blog.csdn.net/qq_37782336/article/details/105160785