总结
P4075 [SDOI2016]模式字符串
题目描述
给出n个结点的树结构T，其中每一个结点上有一个字符，这里我们所说的字符只考虑大写字母A到Z，再给出长度为m的模式串s，其中每一位仍然是A到z的大写字母。
Alice希望知道，有多少对结点
这里结点对
所谓模式串的重复，是将若干个模式串S依次相接（不能重叠)。例如当S=PLUS的时候，重复两次会得到PLUSPLUS，重复三次会得到PLUSPLUSPLUS，同时要注恿，重复必须是整数次的。例如当S=XYXY时，因为必须重复整数次，所以XYXYXY不能看作是S重复若干次得到的。
输入格式
每一个数据有多组测试，
第一行输入一个整数C，表示总的测试个数。
对于每一组测试来说：
第一行输入两个整数，分别表示树T的结点个数n与模式长度m。结点被依次编号为1到n，
之后一行，依次给出了n个大写字母（以一个长度为n的字符串的形式给出），依次对应树上每一个结点上的字符（第i个字符对应了第i个结点)。
之后n-1行，每行有两个整数u和v表示树上的一条无向边，之后一行给定一个长度为m的由大写字母组成的字符串，为模式串S。
输出格式
给出C行，对应C组测试。
每一行输出一个整数，表示有多少对节点
输入输出样例
输入 #1
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输出 #1
5说明/提示
1<=C<=10，3<=∑N<=1000000，3<=∑M<=1000000
看到这题 ，弱弱的我的第一反应就是 \(n^3\) 暴力大枚举 ， 之后幸亏迷途知返，想了一个\(n^2\) 的。。。。。
就不讨论我的弱了，
正解
点分治 ， 用两个数组 \(f , g\) 分别表示前缀 ， 后缀长为 i 的点的数量 ， 之后就是按照点分治的模板 ， 改改就好了 ， 至于怎么判断是否相同，hash就行了。
我还是比较弱 ， 这种题也就算个点分治的板子 ， 以后得多练一些点分治的题。
#include
#include
#include
using namespace std;
const int N = 1010000;
typedef unsigned long long ull;
inline int read()
{
register int x = 0; register char c = getchar();
while(c < '0' || c > '9') c = getchar();
while(c >= '0' && c <= '9') x = (x << 3) + (x << 1) + c - '0' , c = getchar();
return x;
}
int n , m , cnt , root , all , ans;
int d[N] , head[N] , Max[N] , siz[N] , vis[N] , sf[N] , sg[N] , f[N] , g[N];
char a[N] , b[N];
ull t1[N] , t2[N] , bs[N];
struct edge{int v , nex; } e[N<<1];
inline void add(int u , int v) { e[++cnt].v = v; e[cnt].nex = head[u]; head[u] = cnt; return ; }
void getroot(int x , int fa)
{
siz[x] = 1; Max[x] = 0;
for(int i = head[x] , v ; i ; i = e[i].nex)
{
v = e[i].v; if(v == fa || vis[v]) continue;
getroot(v , x); siz[x] += siz[v];
Max[x] = max(Max[x] , siz[v]);
}
Max[x] = max(Max[x] , all - siz[x]);
if(Max[root] > Max[x]) root = x;
return ;
}
int getdep(int x , int fa , int dep , ull hs)
{
hs = hs * 1331 + a[x]; int tmp = 1;
if(hs == t1[dep]) f[(dep - 1) % m + 1]++ , ans += sg[m - (dep - 1) % m];
if(hs == t2[dep]) g[(dep - 1) % m + 1]++ , ans += sf[m - (dep - 1) % m];
for(int i = head[x] , v; i ; i = e[i].nex)
{
v = e[i].v; if(v == fa || vis[v]) continue;
tmp = max(tmp , getdep(v , x , dep + 1 , hs) + 1);
}
return tmp;
}
void dfs(int x)
{
sf[1] = sg[1] = 1; vis[x] = 1; int tmp = 0;
for(int i = head[x] , v; i ; i = e[i].nex)
{
v = e[i].v; if(vis[v]) continue;
int k = min(m , getdep(v , x , 2 , a[x]) + 1); tmp = max(tmp , k);
for(int j = 1 ; j <= k ; ++j) sf[j] += f[j] , sg[j] += g[j] , f[j] = g[j] = 0;
}
for(int i = 1 ; i <= tmp ; ++i) sf[i] = sg[i] = 0;
for(int i = head[x] ; i ; i = e[i].nex)
if(!vis[e[i].v])
{
all = siz[e[i].v]; root = 0;
getroot(e[i].v , x); dfs(root);
}
return ;
}
void solve()
{
n = read(); m = read();
scanf("%s",a + 1);
for(int i = 1 , u , v ; i < n ; ++i)
{
u = read(); v = read();
add(u , v); add(v , u);
}
scanf("%s",b + 1); bs[0] = 1;
for(int i = 1 ; i <= n ; ++i) // hash
{
bs[i] = bs[i-1] * 1331;
t1[i] = t1[i-1] + bs[i-1] * b[(i-1) % m + 1];
t2[i] = t2[i-1] + bs[i-1] * b[m - (i-1) % m];
}
Max[0] = 1e9; root = 0; all = n; getroot(1 , 0); dfs(root);
printf("%d\n" , ans);
for(int i = 1 ; i <= n ; ++i) head[i] = 0 , vis[i] = 0; cnt = 0; ans = 0;
return ;
}
int main()
{
// freopen("A.in" , "r" , stdin);
// freopen("A.out" , "w" , stdout);
int T = read(); while(T --) solve();
fclose(stdin); fclose(stdout); return 0;
}
/*
2
11 4
IODSSDSOIOI
1 2
2 3
3 4
1 5
5 6
6 7
3 8
8 9
6 10
10 11
SDOI
11 4
IODSSDSOIOI
1 2
2 3
3 4
1 5
5 6
6 7
3 8
8 9
6 10
10 11
SDOI
*/