leetcode 745. Prefix and Suffix Search 字典树（前缀树）

傷城~ 2022-06-03 09:56 131阅读 0赞

Given many words, words\[i\] has weight i.

Design a class WordFilter that supports one function, WordFilter.f(String prefix, String suffix). It will return the word with given prefix and suffix with maximum weight. If no word exists, return -1.

Examples:  
Input:  
WordFilter(\[“apple”\])  
WordFilter.f(“a”, “e”) // returns 0  
WordFilter.f(“b”, “”) // returns -1  
Note:  
words has length in range \[1, 15000\].  
For each test case, up to words.length queries WordFilter.f may be made.  
words\[i\] has length in range \[1, 10\].  
prefix, suffix have lengths in range \[0, 10\].  
words\[i\] and prefix, suffix queries consist of lowercase letters only.

本题题意很简单其实考察的就是前缀树，就这么做吧

代码如下：

#include <iostream>
    #include <vector>
    #include <map>
    #include <unordered_map>
    #include <set>
    #include <unordered_set>
    #include <queue>
    #include <stack>
    #include <string>
    #include <climits>
    #include <algorithm>
    #include <sstream>
    #include <functional>
    #include <bitset>
    #include <numeric>
    #include <cmath>
    #include <regex>
    
    using namespace std;
    
    
    struct Trie 
    {
        vector<int> words; // index of words
        vector<Trie *> children;
        Trie() 
        {
            children = vector<Trie *>(26, nullptr);
        }
        // Thanks to @huahualeetcode for adding this in case of memory leak
        ~Trie()
        {
            for (int i = 0; i < 26; i++)
            {
                if (children[i]) 
                    delete children[i];
            }
        }
    
        void add(const string &word, size_t begin, int index) 
        {
            words.push_back(index);
            if (begin < word.length())
            {
                if (!children[word[begin] - 'a']) 
                    children[word[begin] - 'a'] = new Trie();
    
                children[word[begin] - 'a']->add(word, begin + 1, index);
            }
        }
    
        vector<int> find(const string &prefix, size_t begin)
        {
            if (begin == prefix.length())
                return words;
            else 
            {
                if (!children[prefix[begin] - 'a'])
                    return{};
                else
                    return children[prefix[begin] - 'a']->find(prefix, begin + 1);
            }
        }
    };
    
    
    
    class WordFilter 
    {
    public:
        Trie *forwardTrie, *backwardTrie;
    
        WordFilter(vector<string> words)
        {
            forwardTrie = new Trie();
            backwardTrie = new Trie();
            for (int i = 0; i < words.size(); i++)
            {
                string word = words[i];
                forwardTrie->add(word, 0, i);
                reverse(word.begin(), word.end());
                backwardTrie->add(word, 0, i);
            }
        }
    
        int f(string prefix, string suffix)
        {
            auto forwardMatch = forwardTrie->find(prefix, 0);
            reverse(suffix.begin(), suffix.end());
            auto backwardMatch = backwardTrie->find(suffix, 0);
    
            // search from the back
            auto fIter = forwardMatch.rbegin(), bIter = backwardMatch.rbegin();
            while (fIter != forwardMatch.rend() && bIter != backwardMatch.rend()) 
            {
                if (*fIter == *bIter)
                    return *fIter;
                else if (*fIter > *bIter)
                    fIter++;
                else 
                    bIter++;
            }
            return -1;
        }
    };
    
    /**
    * Your WordFilter object will be instantiated and called as such:
    * WordFilter obj = new WordFilter(words);
    * int param_1 = obj.f(prefix,suffix);
    */