[HDOJ 4929] Another Letter Tree [LCA+动态规划]

给定一颗树，每个节点上有一个字母，再给定字符串S0，若干次询问，问从u到的路上构成的字符串包含S0这个子序列多少次。

数据范围：树的规模和询问次数均为10^5，S0的长度不超过30。

通过一次dfs计算出每个节点到根的，包含S0的第i位到第j位这个字符串为子序列多少次。其实是个DP...但是因为每个点是由它的父亲推过来的而不是它的儿子..所以不能算做树形dp...

查询的时候先求出lca，再枚举从从u到lca，从lca到r分别有S0的多少位。计算从u到lca的时候用从u到根的减去从u到lca的一小段乘lca到根的另一段，也是枚举。感觉上也算是个dp吧..

复杂度是S0长度的平方。

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int mod=10007;

struct Node {
	int fe,h;
	int f[20];
	char v;
	int dp[30][30];
};
struct Edge {
	int t,ne;
};

Node a[50001];
Edge b[100000];
char s[31];
int n,bp,ls;

void putedge(int x,int y) {
	b[bp].t=y;
	b[bp].ne=a[x].fe;
	a[x].fe=bp++;
}
void dfs(int i,int f) {
	a[i].f[0]=f;
	for (int j=1;j<20;j++) 
		if (a[i].f[j-1]==0) a[i].f[j]=0;
		else a[i].f[j]=a[a[i].f[j-1]].f[j-1];
	a[i].h=a[f].h+1;
	for (int j=0;j<ls;j++) {
		if (a[i].v==s[j]) {
			a[i].dp[j][j]=(a[f].dp[j][j]+1)%mod;
			for (int k=0;k<j;k++) a[i].dp[j][k]=(a[f].dp[j][k]+a[f].dp[j-1][k])%mod;
			for (int k=j+1;k<ls;k++) a[i].dp[j][k]=(a[f].dp[j][k]+a[f].dp[j+1][k])%mod;
		} else {
			for (int k=0;k<ls;k++) a[i].dp[j][k]=a[f].dp[j][k];
		}
	}
	for (int j=a[i].fe;j!=-1;j=b[j].ne) {
		if (f!=b[j].t) dfs(b[j].t,i);
	}
}
int lca(int x,int y) {
	if (a[x].h<a[y].h) swap(x,y);
	for (int i=19;i>=0;i--) {
		if (a[a[x].f[i]].h>=a[y].h) x=a[x].f[i];
	}
	if (x==y) return x;
	for (int i=19;i>=0;i--) {
		if (a[x].f[i]!=a[y].f[i]) {
			x=a[x].f[i];
			y=a[y].f[i];
		}
	}
	return a[x].f[0];
}
int qx[30];
int qy[30];
void calx(int x,int ac) {
	for (int i=0;i<ls;i++) {
		qx[i]=a[x].dp[0][i]-a[ac].dp[0][i];
		for (int j=0;j<i;j++) {
			qx[i]-=qx[j]*a[ac].dp[j+1][i]%mod;
		}
		qx[i]=(qx[i]%mod+mod)%mod;
	}
}
void caly(int y,int ac) {
	for (int i=ls-1;i>=0;i--) {
		qy[i]=a[y].dp[ls-1][i]-a[ac].dp[ls-1][i];
		for (int j=ls-1;j>i;j--) {
			qy[i]-=qy[j]*a[ac].dp[j-1][i]%mod;
		}
		qy[i]=(qy[i]%mod+mod)%mod;
	}
}
int calAns(int x,int y,int ac) {
	int acf=a[ac].f[0];
	if (x==y) {
		if (a[x].v==s[0]&&ls==1) return 1;
		else return 0;
	} else if (a[x].h==a[ac].h) {
		caly(y,acf);
		return qy[0];
	} else if (a[y].h==a[ac].h) {
		calx(x,acf);
		return qx[ls-1];
	} else {
		calx(x,acf);
		caly(y,ac);
		int ans=qx[ls-1]+qy[0];
		for (int i=ls-2;i>=0;i--) {
			ans=(ans+qx[i]*qy[i+1])%mod;
		}
		return ans;
	}
}

int main() {
	int t,q,i;
	scanf("%d",&t);
	while (scanf("%d%d",&n,&q)!=EOF) {
		bp=0;
		for (i=1;i<=n;i++) {
			a[i].fe=-1;
		}
		for (i=1;i<n;i++) {
			int x,y;
			scanf("%d%d",&x,&y);
			putedge(x,y);
			putedge(y,x);
		}
		for (i=1;i<=n;i++) scanf(" %c",&a[i].v);
		scanf("%s",s);
		ls=strlen(s);
		a[0].h=0;
		memset(a[0].dp,0,sizeof(a[0].dp));
		dfs(1,0);
		//printf("Build Complete!!\n");
		for (i=0;i<q;i++) {
			int x,y;
			scanf("%d%d",&x,&y);
			int ac=lca(x,y);
			//printf("Query: %d %d -- lca=%d\n",x,y,ac);
			int ans=calAns(x,y,ac);
			printf("%d\n",ans);
		}
	}
	return 0;
}