牛客——接头密匙

题目链接:牛客--接头密匙

该题是一个很显然的前缀树问题,只需要构建a中所有数组对应的前缀树,之后求b所处前缀个数即可。关于前缀树的构建,可以观看左老师算法讲解045的视频,简单来讲就是用特殊字符将实际数据隔开,同时将实际数据离散化。示例题解如下:

cpp 复制代码
class Solution {
  private:
    static constexpr int N = 1e6 + 1;
    vector<vector<int>> trie = vector<vector<int>>(N,
                               vector<int>(12));   // 0 - 9 & '-' & '#'
    vector<int> passArr = vector<int>(N);
    vector<int> endArr = vector<int>(N);
    int cnt;

    int getPath(char c) {
        if (c == '-') {
            return 10;
        }
        if (c == '#') {
            return 11;
        }
        return c - '0';
    }

    void buildTrie() {
        cnt = 1;
    }

    void reBuildTrie() {
        for (int i = 0; i < cnt; ++i) {
            fill(&trie[i][0], &trie[i][0] + 12, 0);
            passArr[i] = 0;
            endArr[i] = 0;
        }
    }

    void insert(const string& word) {
        int cur = 1, path;
        passArr[cur]++;
        for (char c : word) {
            path = getPath(c);
            if (trie[cur][path] == 0) {
                trie[cur][path] = ++cnt;
            }
            cur = trie[cur][path];
            passArr[cur]++;
        }
        endArr[cur]++;
    }

    int prefixCount(const string& word) {
        int cur = 1, path;
        for (char c : word) {
            path = getPath(c);
            if (trie[cur][path] == 0) {
                return 0;
            }
            cur = trie[cur][path];
        }
        return passArr[cur];
    }

  public:
    vector<int> countConsistentKeys(vector<vector<int> >& b,
                                    vector<vector<int> >& a) {
        // 建立前缀树
        buildTrie();
        for (int i = 0; i < a.size(); ++i) {
            // a[i] = vector<int>
            string s = "";
            for (int j = 1; j < a[i].size(); j++) {
                s = s + std::to_string(a[i][j] - a[i][j - 1]) + "#";
            }
            insert(s);
        }

        vector<int> ans;
        // 搜索前缀树
        for (int i = 0; i < b.size(); ++i) {
            // b[i] = vector<int>
            string pre = "";
            for (int j = 1; j < b[i].size(); ++j) {
                pre = pre + std::to_string(b[i][j] - b[i][j - 1]) + "#";
            }
            ans.push_back(prefixCount(pre));
        }

        reBuildTrie();

        return ans;
    }
};
相关推荐
拾光Ծ几秒前
【C++】STL有序关联容器的双生花:set/multiset 和 map/multimap 使用指南
数据结构·c++·算法
澄澈i44 分钟前
设计模式学习[20]---桥接模式
c++·学习·设计模式·桥接模式
我星期八休息1 小时前
C++异常处理全面解析:从基础到应用
java·开发语言·c++·人工智能·python·架构
2401_841495642 小时前
【数据结构】汉诺塔问题
java·数据结构·c++·python·算法·递归·
Q741_1472 小时前
C++ 位运算 高频面试考点 力扣137. 只出现一次的数字 II 题解 每日一题
c++·算法·leetcode·面试·位运算
爱编程的鱼2 小时前
Python 与 C++、C 语言的区别及选择指南
c语言·开发语言·c++
奔跑吧邓邓子2 小时前
【C++实战(78)】解锁C++ 大数据处理:从并行到分布式实战
c++·分布式·实战·并发·大数据处理
浔川python社3 小时前
《C++ 实际应用系列》第二部分:内存管理与性能优化实战
c++
liulilittle3 小时前
OPENPPP2 静态隧道链路迁移平滑(UDP/IP)
开发语言·网络·c++·网络协议·tcp/ip·udp·通信