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