串套串

思路

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

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;
    }
};
全部评论

相关推荐

研一开学九月份速成的Java,项目是苍穹外卖和黑马点评,算法基础不好,八股文较为熟练,想找份小厂日常实习,希望牛友们给点意见,蟹蟹啦
求offer的花生米很聪敏:三个月学了这么多?spring springmvc mybatis springboot jvm juc,还做完了两个项目,还熟悉八股,会点算法。卧槽,我该反思了。我暑假开始的,就做了外卖,spring springmvc boot 那些原理好多都忘了,还在刷 jvm 视频,八股和算法也没开始
点赞 评论 收藏
分享
11-27 12:36
已编辑
门头沟学院 前端工程师
Apries:这个阶段来说,很厉害很厉害了,不过写的简历确实不是很行,优势删掉吧,其他的还行
点赞 评论 收藏
分享
M_bao:简历排版换一下吧,第二个项目换了吧,咱门双非学历本来就不行还用这种项目太掉分了,300沟通一个要简历你打招呼也有问题。
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务