[codeforces291E]Tree-String Problem(hash+倍增)

6 篇文章 0 订阅

题目:

我是超链接

题意:

给定一棵边上有字母的有根树,根为1,用点对(v,x)表示v节点通往父亲的那条边上从上往下数第x个(字符串从0开始),给出一个字符串t,询问有多少个对点对——–开始点和结束点——–使之间包括构成的字符串形成t(包括开始点和结束点)

题解:

一个思路是把这棵树边上的每个字符当成一个点,求出这个点到他len(t)高度父亲的hash值,判断是否与给定hash值相等。
Q:如何建这棵树? A:先把n个点弄成空点,然后字符串上的点全都堆到后面
Q:如何计算定高度hash值? A:可以先计算到根节点的hash,然后减去半山腰的hash
对于hash进一步的理解:hash[x]-hash[fa]*mul指的是x到fa的这一串字符
最后一定不要忘了把进制变成一个奇怪的数字!!!!!

代码:

#include <cstdio>
#include <iostream>
#include <cstring>
#define ull unsigned long long
using namespace std;
const int N=5e5; 
const int sz=19;
const int base=130000001ull;
char st[N];
int tot,nxt[N*2],point[N],v[N*2],cnt=0,ans,h[N];
int f[N][sz],len,bel[N];
ull mul,Mo,Ha[N];
void addline(int x,int y){++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y;}
void dfs(int x,int fa)
{
    Ha[x]=Ha[fa]*base+bel[x];
    for (int i=1;i<sz;i++) f[x][i]=f[f[x][i-1]][i-1];
    for (int i=point[x];i;i=nxt[i])
      {
        f[v[i]][0]=x;
        dfs(v[i],x);
      }
}
int find(int x)
{
    for (int i=0;i<sz;i++)
      if ((len>>i)&1) x=f[x][i];
    return x;
}
void answer(int x,int fa)
{
    int father=find(x);
    if (father)
    {
        ull wh=Ha[x]-Ha[father]*mul;
        if (wh==Mo) ans++;
    }
    for (int i=point[x];i;i=nxt[i]) answer(v[i],x);
}
int main()
{
    int n,i,j;
    scanf("%d",&n);cnt=n;
    for (i=2;i<=n;i++)
    {
        int x;scanf("%d%s",&x,&st);
        int ll=strlen(st);
        if (ll==1){bel[i]=st[0];addline(x,i);continue;}
        bel[++cnt]=st[0];addline(x,cnt);
        for (j=1;j<ll-1;j++) bel[++cnt]=st[j],addline(cnt-1,cnt);
        bel[i]=st[ll-1];addline(cnt,i);
    }
    scanf("%s",st);len=strlen(st);
    mul=1;for (i=0;i<len;i++) Mo=Mo*base+st[i],mul*=base;
    dfs(1,0);
    answer(1,0);
    printf("%d",ans);
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值