UVa 1401 字典树+DP

白书上的题目真心有点无奈啊。这个感觉单单字典树其实还好的,但是一加了个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;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值