题目描述
单词数组 words 的 有效编码 由任意助记字符串 s 和下标数组 indices 组成,且满足:
words.length == indices.length
助记字符串 s 以 ‘#‘ 字符结尾
对于每个下标 indices[i] ,s 的一个从 indices[i] 开始、到下一个 ‘#‘ 字符结束(但不包括 ‘#‘)的 子字符串 恰好与 words[i] 相等
给你一个单词数组 words ,返回成功对 words 进行编码的最小助记字符串 s 的长度 。
- 示例
输入:words = ["time", "me", "bell"]
输出:10
解释:一组有效编码为 s = "time#bell#" 和 indices = [0, 2, 5] 。
words[0] = "time" ,s 开始于 indices[0] = 0 到下一个 ‘#‘ 结束的子字符串,如加粗部分所示 "time#bell#"
words[1] = "me" ,s 开始于 indices[1] = 2 到下一个 ‘#‘ 结束的子字符串,如加粗部分所示 "time#bell#"
words[2] = "bell" ,s 开始于 indices[2] = 5 到下一个 ‘#‘ 结束的子字符串,如加粗部分所示 "time#bell#"
解决方案
class TrieNode {
public:
TrieNode() {
for (int i=0;i<26;i++) {
children[i]=nullptr;
}
end=false;
}
bool containsKey(char ch) {
return children[ch-‘a‘]!=nullptr;
}
void put(char ch, TrieNode*node) {
children[ch-‘a‘]=node;
}
TrieNode* get(char ch) {
return children[ch-‘a‘];
}
void setEnd() {
end=true;
}
bool isEnd() {
return end;
}
private:
bool end;
TrieNode *children[26];
};
class Trie {
public:
Trie() {
root=new TrieNode();
}
void insert(string word) {
TrieNode *node=root;
for (char ch:word) {
if (!node->containsKey(ch)) {
node->put(ch, new TrieNode());
}
node=node->get(ch);
}
node->setEnd();
}
private:
TrieNode *root;
};
class Solution {
public:
int minimumLengthEncoding(vector<string>& words) {
int ans=0;
Trie *trie=new Trie();
for (auto &word:words) {
word.reserve();
trie->insert(word);
}
}
};