七月集训（22，23）字典树，有序集合

1.LeetCode:1418. 点菜展示表

原题链接

        给你一个数组 orders，表示客户在餐厅中完成的订单，确切地说， orders[i]=[customerNamei,tableNumberi,foodItemi] ，其中 customerNamei 是客户的姓名，tableNumberi 是客户所在餐桌的桌号，而 foodItemi 是客户点的餐品名称。

        请你返回该餐厅的 点菜展示表 。在这张表中，表中第一行为标题，其第一列为餐桌桌号 “Table” ，后面每一列都是按字母顺序排列的餐品名称。接下来每一行中的项则表示每张餐桌订购的相应餐品数量，第一列应当填对应的桌号，后面依次填写下单的餐品数量。

        注意：客户姓名不是点菜展示表的一部分。此外，表中的数据行应该按餐桌桌号升序排列。

        示例 1：

        输入：orders = [[“David”,“3”,“Ceviche”],[“Corina”,“10”,“Beef Burrito”],[“David”,“3”,“Fried Chicken”],[“Carla”,“5”,“Water”],[“Carla”,“5”,“Ceviche”],[“Rous”,“3”,“Ceviche”]]

        输出：[[“Table”,“Beef Burrito”,“Ceviche”,“Fried Chicken”,“Water”],[“3”,“0”,“2”,“1”,“0”],[“5”,“0”,“1”,“0”,“1”],[“10”,“1”,“0”,“0”,“0”]]

        示例 2：

        输入：orders = [[“James”,“12”,“Fried Chicken”],[“Ratesh”,“12”,“Fried Chicken”],[“Amadeus”,“12”,“Fried Chicken”],[“Adam”,“1”,“Canadian Waffles”],[“Brianna”,“1”,“Canadian Waffles”]]

        输出：[[“Table”,“Canadian Waffles”,“Fried Chicken”],[“1”,“2”,“0”],[“12”,“0”,“3”]]

        示例 3：

        输入：orders = [[“Laura”,“2”,“Bean Burrito”],[“Jhon”,“2”,“Beef Burrito”],[“Melissa”,“2”,“Soda”]]

        输出：[[“Table”,“Bean Burrito”,“Beef Burrito”,“Soda”],[“2”,“1”,“1”,“1”]]

        提示：

        1 <= orders.length <= 5 * 10^4

        orders[i].length == 3

        1 <= customerNamei.length, foodItemi.length <= 20

        customerNamei 和 foodItemi 由大小写英文字母及空格字符 ’ ’ 组成。

        tableNumberi 是 1 到 500 范围内的整数。

        本题中顾客姓名是完全没用的信息，我们只需要利用一个set记录出现过的菜单并利用一个map统计每个餐桌上点过菜单的数目，最后遍历即可。

class Solution {
    int str_toint(const string& s){
        int ans=0;
        for(int i=0;i<s.size();++i){
            ans=ans*10+s[i]-'0';
        }
        return ans;
    }
    string to_str(int x){
        if(!x){
            return "0";
        }
        string ans;
        while(x){
            ans.push_back(x%10+'0');
            x/=10;
        }
        reverse(ans.begin(),ans.end());
        return ans;
    }
public:
    vector<vector<string>> displayTable(vector<vector<string>>& orders) {
        map<string,int> map[501];
        set<string> s;
        int i;
        vector<vector<string>> ans;
        for(i=0;i<orders.size();++i){
            int tablenum=str_toint(orders[i][1]);
            string food=orders[i][2];
            map[tablenum][food]++;
            s.insert(food);
        }
        vector<string> ttmp;
        ttmp.push_back("Table");
        for(auto & it:s){
            ttmp.push_back(it);
        }
        ans.push_back(ttmp);
        for(i=1;i<=500;++i){
            vector<string> tmp;
            if(map[i].size()){
                tmp.push_back(to_str(i));
                for(auto it=s.begin();it!=s.end();++it){
                    tmp.push_back(to_str(map[i][*it]));
                }
                ans.push_back(tmp);
            }
        }
        return ans;
    }
};

2.LeetCode:1032. 字符流

原题链接

        设计一个算法：接收一个字符流，并检查这些字符的后缀是否是字符串数组 words 中的一个字符串。

        例如，words = [“abc”, “xyz”] 且字符流中逐个依次加入 4 个字符 ‘a’、‘x’、‘y’ 和 ‘z’ ，你所设计的算法应当可以检测到 “axyz” 的后缀 “xyz” 与 words 中的字符串 “xyz” 匹配。

        按下述要求实现 StreamChecker 类：

        StreamChecker(String[] words) ：构造函数，用字符串数组 words 初始化数据结构。

        boolean query(char letter)：从字符流中接收一个新字符，如果字符流中的任一非空后缀能匹配 words 中的某一字符串，返回 true ；否则，返回 false。

        示例：

        输入：

        [“StreamChecker”, “query”, “query”, “query”, “query”, “query”, “query”, “query”, “query”, “query”, “query”, “query”, “query”]

        [[[“cd”, “f”, “kl”]], [“a”], [“b”], [“c”], [“d”], [“e”], [“f”], [“g”], [“h”], [“i”], [“j”], [“k”], [“l”]]

        输出：

        [null, false, false, false, true, false, true, false, false, false, false, false, true]

        提示

        1 <= words.length <= 2000

        1 <= words[i].length <= 200

        words[i] 由小写英文字母组成

        letter 是一个小写英文字母

        最多调用查询 4 * 1e4 次

        反向建立字典树，并反向查找即可，在查找的过程中遇到了某个结点的isend为true即可返回true。

struct Tnode{
    Tnode* next[26];
    bool end;
};
class Trie{
public:
    Tnode* root;
    Trie(){
        root=new Tnode();
        memset(root,0,sizeof(root->next));
        root->end=false;
    }

    void insert(const string& s){
        Tnode* cur=root;
        for(int i=s.size()-1;i>=0;--i){
            int idx=s[i]-'a';
            if(!cur->next[idx]){
                cur->next[idx]=new Tnode();
            }
            cur=cur->next[idx];
        }
        cur->end=true;
    }

    bool search(const string & s){
        Tnode *cur=root;
        for(int i=s.size()-1;i>=0;--i){
            int idx=s[i]-'a';
            if(!cur->next[idx]){
                return false;
            }
            cur=cur->next[idx];
            if(cur->end){
                return true;
            }
        }
        return false;
    }
};
class StreamChecker {
    Trie tree;
    string s;
public:
    StreamChecker(vector<string>& words) {
        for(auto & s:words){
            tree.insert(s);
        }
    }
    
    bool query(char letter) {
        s+=letter;
        return tree.search(s);
    }
};