给你若干给字符串,再给你一个m,问长度是m的字符串中包含给定字符串的数量mod 10007是多少
这个拿过来啥思路也没有,后来还是看了题解,才知道,原来,原来。。。。那个带fail的Trie还可以搞别的
网上的大多数都是26^m-补集做的,麻烦啊,网上有一个非洲猴的blog写的挺好的
设f[i][j][k]表示什么呢,表示i:0..1 j:0..m k:0...size
表示i状态,匹配到j,在Trie上是k的数量
f[i | val[next[c]]][j + 1][next[c]] += f[i][j][k];对就是这样
我已经把next[c] == NULL的都变成fail了,
提醒按照lrj的白书学的AC自动机的同学们,那个last数组在匹配的所有时候都要循环枚举
提醒我自己当k等于0是不是这个位置是无,而是一个未匹配的点,因为计算last和dp时用last耗时代码量费啊
#include <cstdio> #include <cstdlib> #include <cstring> #include <queue> #include <algorithm> using namespace std; struct AC{ int ch[100010][30]; int val[100010]; char s[100010]; int f[100010]; int last[100010]; int g[2][200][10010]; int size; int len; inline int num(char x){ return x - 'A'; } inline void init(){ memset(ch, 0, sizeof(ch)); memset(val, 0, sizeof(val)); size = 0; return; } inline void insert(){ int u = 0; for(int i = 0; i < len; i ++){ int c = num(s[i]); if(!ch[u][c]){ ch[u][c] = ++ size; } u = ch[u][c]; } val[u] = 1; return; } inline void getfail(){ queue<int> Q; for(int i = 0; i < 26; i ++) { if(ch[0][i]) { Q.push(ch[0][i]); } } while(!Q.empty()){ int r = Q.front(); Q.pop(); for(int i = 0; i < 26; i ++){ int u = ch[r][i]; if(!u){ ch[r][i] = ch[f[r]][i]; continue; } Q.push(u); int v = f[r]; while(v && !ch[v][i]) v = f[v]; f[u] = ch[v][i]; } } return; } inline int dp(int m){ for(int i = 0; i <= size; i ++){ if(!val[i]){ int j = f[i]; while(j){ if(val[j] == 1){ val[i] = 1; break; } j = f[j]; } } } g[0][0][0] = 1; for(int i = 0; i <= 1; i ++){ for(int j = 0; j < m; j ++){ for(int k = 0; k <= size; k ++){ for(int c = 0; c < 26; c ++){ g[i | val[ch[k][c]]][j + 1][ch[k][c]] += g[i][j][k]; g[i | val[ch[k][c]]][j + 1][ch[k][c]] %= 10007; } } } } int ret = 0; for(int k = 0; k <= size; k ++) { ret += g[1][m][k]; ret %= 10007; } return ret; } } wt; int main(){ int N, M; scanf("%d%d", &N, &M); wt.init(); for(int i = 1; i <= N; i ++){ scanf("%s", wt.s); wt.len = strlen(wt.s); //if(wt.len > M) continue; wt.insert(); } wt.getfail(); printf("%d\n", wt.dp(M)); return 0; }