题目:
题意:
给定一棵边上有字母的有根树,根为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);
}