串套串

思路

将所有的字符串插到字典树中,将串的结束点权值加一。
再次遍历所有字符串,统计其在字典树路径上的权值和。

Code

class Solution {
#define maxn 2200010
private:
    int trie[maxn][26], pn = 0;
    int ed[maxn];

public:
    /**
     * 返回新集合的种类数
     * @param n int整型 集合大小
     * @param m int整型 限制数量
     * @param limit Point类vector 不能同时出现的(u,v)集合
     * @return int整型
     */

    void insert(string s) {
        int p = 0;
        for (int ch, i = 0; s[i]; ++i) {
            ch = s[i] - 'a';
            if (!trie[p][ch]) trie[p][ch] = ++pn;
            p = trie[p][ch];
        }
        ++ed[p];
    }
    int query(string s) {
        int p = 0, ans = 0;
        for (int ch, i = 0; s[i]; ++i) {
            ch = s[i] - 'a';
            p = trie[p][ch];
            ans += ed[p];
        }
        ans -= ed[p];
        return ans;
    }
    vector<int> solve(int n, vector<string>& str_list) {
        // write code here
        memset(trie, 0, sizeof trie);
        memset(ed, 0, sizeof ed);
        pn = 0;

        for (int i = 0; i < n; ++i) {
            insert(str_list[i]);
        }
        vector<int> ans;
        for (int i = 0; i < n; ++i) {
            ans.push_back(query(str_list[i]));
        }
        return ans;
    }
};
全部评论

相关推荐

尊尼获获:闺蜜在哪?
点赞 评论 收藏
分享
12-01 12:34
已编辑
广东工业大学 Java
如题,fw🐭🐭,加上准备的太晚,大三上已找不到日常实习,导致连锁反应,下学期的暑期实习找不到好的实习,导致秋招找不到中大厂,现在是中小厂Java还有考公的选择,由于有些中小厂工作强度比肩大厂,钱还少,感觉不如考公如果🐮u们是我现在这种情况,会怎么选?
负债的混子:关注你一段时间了,突然发现你头像名字都改了,想必是这段时间压力很大。关于就业还是考公的选择,就像很多牛友说的:不要美化自己没走过的路。你现在想往互联网发展,发现这条路很难走,然后想往考公发展,但是你没走过考公这条路,所以你不知道这条路的压力如何。你今年大三了,还有时间给你做选择,我希望你能够尽快的决定自己的方向,然后一条路走到黑,而不是在这里徘徊,每个人的道路是不一样的,你无法复刻别人的路,你能做的就是尽力的完善自己。 最后,我想说的是,加油,陌生人!
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务