zoukankan      html  css  js  c++  java
  • 【洛谷 P1666】 前缀单词 (Trie)

    题目链接

    考试时暴搜50分。。。其实看到“单词”,“前缀”这种字眼时就要想到(Trie)的,哎,我太蒻了。
    以一个虚点为根,建一棵(Trie),然后(dfs)
    以当前点为根的答案就是(Ans_u=(prod_{ ext{v是u的子树}}Ans_v)+ ext{有单词以这个点结尾 ? 1 : 0}),乘法原理嘛,如果有单词在这里结尾,那么就多一种情况:选这个单词且不选所有子树。
    答案就是(Ans_{ ext{根}})

    #include <cstdio>
    #include <iostream>
    #include <cstring>
    using namespace std;
    #define Open(s) freopen(s".in","r",stdin);freopen(s".out","w",stdout);
    #define Close fclose(stdin);fclose(stdout);
    int n;
    char a[60];
    struct Trie{
        Trie *son[27];
        int p;
    }*root = new Trie();
    inline void Insert(){
        int len = strlen(a + 1);
        Trie *now = root;
        for(int i = 1; i <= len; ++i){
           if(now->son[a[i] - 'a'] == NULL)
             now->son[a[i] - 'a'] = new Trie(), now->son[a[i] - 'a']->p = 0;
           now = now->son[a[i] - 'a'];
        }
        now->p = 1;
    }
    long long dp(Trie *now){
        long long sum = 1;
        for(int i = 0; i < 26; ++i)
           if(now->son[i] != NULL){
             sum *= dp(now->son[i]);
           }
        return sum + now->p;
    }
    int main(){
    	Open("prefix");
    	scanf("%d", &n);
    	for(int i = 1; i <= n; ++i)
    		scanf("%s", a + 1), Insert();
    	printf("%lld
    ", dp(root));
    	return 0;
    }
    
    
  • 相关阅读:
    第二个周六——3.9
    女王节,很开心——3.8
    女生节——3.7
    尴尬的一批——3.6
    周二——3.5
    周一——3.4
    Java基本语法_循环练习系列(二)——万年历
    Java基本语法_循环练习系列(一)——模拟双色球
    《剩女郎》的艺术魅力
    剩女郎剧评
  • 原文地址:https://www.cnblogs.com/Qihoo360/p/9668694.html
Copyright © 2011-2022 走看看