题意:有一个长为n(n<= 25) 的字符串,它至少由k个magic word 组成,现在给出m个magic word,求出这个字符串组成的可能种数。
构造AC自动机进行搜索,dp【i】【j】【k】表示字符串长度为i,匹配字典树上的第j个节点,并且已经匹配上k个magic word时的总数。
则转移方程为 (dp【i+1】【j的儿子】【k | j的儿子的状态】 += dp【i】【j】【k】) % mod;
需要注意的是,因为单词可以重复使用,所以单词结尾的fail 指向root 指向的各节点
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include //形如INT_MAX一类的
#define MAX 100005
#define mod 20090717
#define INF 0x7FFFFFFF
#define REP(i,s,t) for(int i=(s);i<=(t);++i)
#define LL long long
#define mem(a,b) memset(a,b,sizeof(a))
#define mp(a,b) make_pair(a,b)
#define L(x) x << 1
#define R(x) x << 1 | 1
# define eps 1e-5
//#pragma comment(linker, "/STACK:36777216") ///传说中的外挂
using namespace std;
struct Trie {
int next[26] ;
int fail ;
int buff ;
void init() {
memset(next,0,sizeof(next)) ;
fail = -1 ;
buff = 0 ;
}
} a[111];
int cnt,n,m,K;
int q[111] , dp[30][111][1 << 10];
char keyword[22];
void insert(char *s,int pos) {
int p = 0 ;
for(int i = 0 ; s[i] ; i ++) {
int t = s[i] - 'a' ;
if(a[p].next[t] == 0) {
a[cnt].init();
a[p].next[t] = cnt ++;
}
p = a[p].next[t] ;
}
a[p].buff = 1 << pos;
}
void ac_bfs() {
int i,head = 0,tail = 0;
q[tail ++] = 0;
while(head < tail) {
int front = q[head ++];
for(i = 0; i < 26 ; i ++) {
if(a[front].next[i] == 0) {
if(front != 0) a[front].next[i] = a[a[front].fail].next[i];
} else {
int p = a[front].fail ;
while(p != -1) {
if(a[p].next[i] != 0) {
a[a[front].next[i]].fail = a[p].next[i] ;
a[a[front].next[i]].buff |= a[a[p].next[i]].buff;//状态合并,因为如果fail指向的节点也是某个magic word的结尾,那么相当于一次找到两个magic word
break ;
}
p = a[p].fail ;
}
if(p == -1) a[a[front].next[i]].fail = 0 ;
q[tail ++] = a[front].next[i] ;
}
}
}
}
int calone(int x) {
int cal = 0;
while(x) {
if(x & 1) cal ++;
x = x >> 1;
}
return cal;
}
void solve() {
dp[0][0][0] = 1;
int total = 1 << m;
for(int i=0; i