原根

最优解

考虑建立一棵字典树,一个字符串为另一个字符串的前缀,当且仅当它对应的节点是另一个节点的祖先。所以我们建立完字典树之后,标记一下每个结点作为字符串的终点几次。最后从根开始往下搜索,当搜索到一个结束位置的时候,如果这个结束位置只代表一个字符串那么就统计入答案,不然就不统计。并且搜到一个结束位置就不用往下搜了,因为下面即使搜到了结束位置也不可能是原根,当前的结束位置必然会是它的祖先。
时间复杂度和空间复杂度都是

int ch[1000005][26];
int is_over[1000005];
int dfs(int u){
    if(is_over[u]){
        return (is_over[u] == 1);
    }
    int ans = 0;
    for(int i = 0; i < 26; ++i) if(ch[u][i]) ans += dfs(ch[u][i]);
    return ans;
}
int solve(int n, vector<string> s){
    int tot = 0; int rt = ++tot;
    memset(ch[rt], 0, sizeof ch[rt]); is_over[rt] = 0;
    for(int i = 0; i < n; ++i){
        int p = rt;
        for(int j = 0; j < s[i].size(); ++j){
            int x = s[i][j] - 'a';
            if(!ch[p][x]) {
                ch[p][x] = ++tot; memset(ch[tot], 0, sizeof ch[tot]);  is_over[tot] = 0;
            }
            p = ch[p][x]; 
        }
        is_over[p]++;
    }
    return dfs(rt);
}
全部评论

相关推荐

想按时下班的大菠萝在...:隔壁学校的,加油多投, 实在不好找可以下个学期开学找,把算法八股准备好,项目有空再换换
投了多少份简历才上岸
点赞 评论 收藏
分享
不要停下啊:大二打开牛客,你有机会开卷了,卷起来,去找课程学习,在牛客上看看大家面试笔试都需要会什么,岗位有什么需求就去学什么,努力的人就一定会有收获,这句话从来都经得起考验,像我现在大三了啥也不会,被迫强行考研,炼狱难度开局,啥也不会,找工作没希望了,考研有丝丝机会
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务