白书上的题目真心有点无奈啊。这个感觉单单字典树其实还好的,但是一加了个DP,我就有点束手无策了,还是思维弱啊,DP这种东西真的是不知道怎么弄了。
引用:
想到递推:开始是从左往右推:dp(i)表示子串s[0, i]的拆分方案数,然后枚举它的每个后缀判断其是否在字典树中,转移方程:dp(i) = { sum(dp(i - len(x))) | x为s[0,
i]的后缀且出现在字典树中 };但是这样时间依然高达O(n^2 * 100),效率还是太低。
上面的方法的瓶颈在于:对于每个子串s[0, i],枚举每一个后缀没有利用到串的连续性,而进行了大量重复的操作。如果是枚举每一个前缀的话就可以在线性时间完成所有的枚举了,那么怎样才能用枚举前缀的方法递推呢。可以想到把从左往右递推改成从右往左,这样就变成了枚举子串s[i, len(s)]的每个前缀了,可以通过在字典树中查找s[i, len(s)]每遇到一个单词节点就对dp(i)进行相应的操作。时间就降为O(n
* 100)。
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=300010;
const int INF=20071027;
const int maxnode=4000010;
char a[maxn],tmp[maxn];
int n;
int ch[maxnode][26];
int sum[maxn];
int val[maxnode];
int sz;
void init()
{
sz=1;
memset(ch[0],0,sizeof(ch[0]));
int n=strlen(a);
for(int i=1;i<=n;i++)
sum[i]=0;
sum[0]=1;
}
int idx(char c)
{
return c-'a';
}
void insert()
{
int u=0,n=strlen(tmp);
for(int i=0;i<n;i++)
{
int c=idx(tmp[i]);
if(!ch[u][c])
{
memset(ch[sz],0,sizeof(ch[sz]));
val[sz]=0;
ch[u][c]=sz++;
}
u=ch[u][c];
}
val[u]=1;
}
void query()
{
int n=strlen(a);
for(int i=1;i<n;i++)
{
int u=0;
for(int j=i;j<n;j++)
{
int c=idx(a[j]);
if(!ch[u][c])
break;
u=ch[u][c];
if(val[u])
sum[j]=(sum[i-1]+sum[j])%INF;
}
}
}
int main()
{
int kase=1;
while(scanf("%s",a+1)!=EOF)
{
a[0]='1';
init();
scanf("%d",&n);
for(int i=0;i<n;i++)
{
scanf("%s",tmp);
insert();
}
query();
int n=strlen(a);
printf("Case %d: %d\n",kase++,sum[n-1]%INF);
}
return 0;
}