[SDOI2016]模式字符串

Description
给出n个结点的树结构T，其中每一个结点上有一个字符，这里我们所说的字符只考虑大写字母A到Z,再给出长度为m的模式串s，其中每一位仍然是A到z的大写字母。Alice希望知道，有多少对结点<u，v>满足T上从u到V的最短路径形成的字符串可以由模式串S重复若干次得到？这里结点对<u，v>是有序的，也就是说<u，v>和<v，u>需要被区分.所谓模式串的重复，是将若干个模式串S依次相接（不能重叠).例如当S=PLUS的时候，重复两次会得到PLUSPLUS，重复三次会得到PLUSPLUSPLUS,同时要注恿，重复必须是整数次的。例如当S=XYXY时，因为必须重复整数次，所以XYXYXY不能看作是S重复若干次得到的。

Input
每一个数据有多组测试，
第一行输入一个整数C，表示总的测试个数。
对于每一组测试来说：
第一行输入两个整数，分别表示树T的结点个数n与模式长度m。结点被依次编号为1到n，之后一行，依次给出了n个大写字母（以一个长度为n的字符串的形式给出），依次对应树上每一个结点上的字符（第i个字符对应了第i个结点).
之后n-1行，每行有两个整数u和v表示树上的一条无向边，之后一行给定一个长度为m的由大写字母组成的字符串，为模式串S。
1<=C<=10,3<=N<=10000003<=M<=1000000

Output
给出C行，对应C组测试。每一行输出一个整数，表示有多少对节点<u,v>满足从u到v的路径形成的字符串恰好是模式串的若干次重复.

Sample Input
1
11 4
IODSSDSOIOI
1 2
2 3
3 4
1 5
5 6
6 7
3 8
8 9
6 10
10 11
SDOI

Sample Output
5

---

字符串匹配，Hash嘛；树上操作，点分治嘛。

首先考虑如何点分，由于路径是由模式串重复多次得来的，我们没必要考虑重复了多少次，因此我们开一个桶，\(hp[i]\)表示当前匹配了模式串的前\(i\)个（在\(\%L\)意义下，\(L\)为其长度），记\(hs[i]\)表示匹配了后\(i\)个（\(pre,suf\)），在dfs过程中可以直接统计答案，然后累加

考虑如何判断dfs过程中的串是否合法，使用Hash，由于dfs过程中可能出现模式串重复多次的情况，因此我们在初始Hash的时候需要将模式串循环倍增

细节可以参考代码

/*program from Wolfycz*/
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define inf 0x7f7f7f7f
using namespace std;
typedef long long ll;
typedef unsigned int ui;
typedef unsigned long long ull;
inline char gc(){
    static char buf[1000000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,1000000,stdin),p1==p2)?EOF:*p1++;
}
inline int frd(){
    int x=0,f=1; char ch=gc();
    for (;ch<'0'||ch>'9';ch=gc())   if (ch=='-')    f=-1;
    for (;ch>='0'&&ch<='9';ch=gc()) x=(x<<3)+(x<<1)+ch-'0';
    return x*f;
}
inline int read(){
    int x=0,f=1; char ch=getchar();
    for (;ch<'0'||ch>'9';ch=getchar())  if (ch=='-')    f=-1;
    for (;ch>='0'&&ch<='9';ch=getchar())    x=(x<<3)+(x<<1)+ch-'0';
    return x*f;
}
inline void print(int x){
    if (x<0)    putchar('-'),x=-x;
    if (x>9)    print(x/10);
    putchar(x%10+'0');
}
const int N=1e6,base=974531;
int pre[(N<<1)+10],now[N+10],child[(N<<1)+10];
int size[N+10],Df[N+10],Up[N+10],Dn[N+10],sup[N+10],sdn[N+10];
ull Suf[N+10],Pre[N+10];
bool vis[N+10];
char s[N+10];
int tot,root,Max,Ans,n,m;
void join(int x,int y){pre[++tot]=now[x],now[x]=tot,child[tot]=y;}
void insert(int x,int y){join(x,y),join(y,x);}
void Get_root(int x,int fa,int sz){
    int res=0; size[x]=1;
    for (int p=now[x],son=child[p];p;p=pre[p],son=child[p]){
        if (son==fa||vis[son])  continue;
        Get_root(son,x,sz);
        size[x]+=size[son];
        res=max(res,size[son]);
    }
    res=max(res,sz-size[x]);
    if (res<Max)    Max=res,root=x;
}
void solve(int x,int fa,ull res,int Deep){
    res=res*base+s[x],Df[x]=Deep;
    if (res==Pre[Deep]) Up[(Deep-1)%m+1]++,Ans+=sdn[m-(Deep-1)%m];
    if (res==Suf[Deep]) Dn[(Deep-1)%m+1]++,Ans+=sup[m-(Deep-1)%m];
    for (int p=now[x],son=child[p];p;p=pre[p],son=child[p]){
        if (son==fa||vis[son])  continue;
        solve(son,x,res,Deep+1);
        Df[x]=max(Df[x],Df[son]);
    }
}
void divide(int x){
    vis[x]=1; int len=0;
    sup[1]=sdn[1]=1;
    for (int p=now[x],son=child[p];p;p=pre[p],son=child[p]){
        if (vis[son])   continue;
        solve(son,0,s[x],2);
        int tmp=min(Df[son],m); len=max(len,tmp);
        for (int i=1;i<=tmp;i++)    sup[i]+=Up[i],sdn[i]+=Dn[i],Up[i]=Dn[i]=0;
    }
    for (int i=1;i<=len;i++)    sup[i]=sdn[i]=0;
    for (int p=now[x],son=child[p];p;p=pre[p],son=child[p]){
        if (vis[son])   continue;
        Max=inf,root=0;
        Get_root(son,0,size[son]);
        divide(root);
    }
}
void init(){
    tot=Ans=0;
    memset(now,0,sizeof(now));
    memset(vis,0,sizeof(vis));
}
int main(){
    for (int Data=read();Data;Data--){
        init(),n=read(),m=read();
        scanf("%s",s+1);
        for (int i=1;i<n;i++){
            int x=read(),y=read();
            insert(x,y);
        }
        static char t[N+10];
        scanf("%s",t+1); ull res=1;
        for (int i=1;i<=n;i++){
            Pre[i]=Pre[i-1]+t[(i-1)%m+1]*res;
            Suf[i]=Suf[i-1]+t[m-(i-1)%m]*res;
            res*=base;
        }
        Max=inf,root=0;
        Get_root(1,0,n);
        divide(root);
        printf("%d\n",Ans);
    }
    return 0;
}

转载于:https://www.cnblogs.com/Wolfycz/p/10214713.html