「bzoj4713」迷失的字符串 - bitset（未AC）

题目大意

    有一棵$n$个节点的大树，上面每条边有一个小写字符。
    对于任意两个不同的点$u,v$，我们可以在树上找到$u$出发到$v$终止的唯一的一条最短路径，并将沿途经过的边上的字符依次写下来，得到一个字符串。
    对于一个字符串，如果存在这样一个点对$(u,v)$，使得它们路径上的字符串与其完全匹配，那么我们就称这个字符串属于这棵树。
    现在有$m$个迷失的字符串，请你写一个程序帮助判断每一条字符串是否属于这棵树。

---

题目分析

开始在想怎么拆正反串后缀自动机。
然后数据范围$30000$，直接上bitset。

考虑只有一个串时，进行如下DP：
设$f(i,j)$表示到达树上第$i$个点，从前往后匹配到第$j$个位置是否可行。
$g(i,j)$是从后往前匹配。

$f(i,j)$可行必须儿子$f(son,j-1)$可行，以及$son\rightarrow i$的字符 $=s_j$。

多个串怎么办？
全部连起来，中间插个间隔符，每个间隔符初始化一下，照样DP即可。

你以为这样就结束了吗？

如果你使用了DFS，恭喜你RE！
改成从下往上拓扑排序。

你以为这样就结束了吗？$\times2$

恭喜你TLE。
因为bitset太大，速度很慢，需要压位，这就是我现在还没A的原因。

---

代码

#include<bits/stdc++.h>

using namespace std;

inline int Get_Int() {
	int num=0,bj=1;
	char x=getchar();
	while(!isdigit(x)) {if(x=='-')bj=-1;x=getchar();}
	while(isdigit(x)) {num=num*10+x-'0';x=getchar();}
	return num*bj;
}

const int maxn=30005;

struct Edge {int from,to,ch;};
vector<Edge> edges[maxn];
bitset<(maxn<<1)> f[maxn],g[maxn],ans,tmpf,tmpg,ch[26];
int father[maxn],dis[maxn],InDegree[maxn];

void AddEdge(int x,int y,char v) {edges[x].push_back((Edge) {x,y,v-'a'});}

void Dfs(int Now,int fa) {
	father[Now]=fa;
	for(Edge &e:edges[Now]) {
		int Next=e.to;
		if(Next==fa)continue;
		Dfs(Next,Now);
		InDegree[Now]++;
		dis[Next]=e.ch;
	}
}

int n,m,l[maxn],r[maxn];
char s[maxn];

void Bfs() {
	queue<int> Q;
	for(int i=1; i<=n; i++)if(!InDegree[i])Q.push(i);
	while(!Q.empty()) {
		int Now=Q.front();
		Q.pop();
		if(Now==1)break;
		int Next=father[Now];
		tmpf=(f[Now]<<1)&ch[dis[Now]];
		tmpg=(g[Now]>>1)&ch[dis[Now]];
		ans|=(f[Next]<<1)&tmpg;
		ans|=tmpf&(g[Next]>>1);
		f[Next]|=tmpf;
		g[Next]|=tmpg;
		if(--InDegree[Next]==0)Q.push(Next);
	}
}

int main() {
	n=Get_Int();
	for(int i=1; i<n; i++) {
		int x=Get_Int(),y=Get_Int();
		char v=' ';
		while(!isalpha(v))v=getchar();
		AddEdge(x,y,v);
		AddEdge(y,x,v);
	}
	m=Get_Int();
	r[0]=1;
	for(int i=1; i<=m; i++) {
		scanf("%s",s+1);
		int len=strlen(s+1);
		l[i]=r[i-1]+1,r[i]=r[i-1]+len;
		for(int j=1; j<=len; j++)ch[s[j]-'a'].set(r[i-1]+j);
		r[i]++; //空字符
	}
	for(int i=0; i<=m; i++)f[0].set(r[i]);
	g[0]=f[0];
	for(int i=1; i<=n; i++)f[i]=g[i]=f[0];
	Dfs(1,0);
	Bfs();
	for(int i=1; i<=m; i++) {
		bool bj=0;
		for(int j=l[i]; j<r[i]; j++)bj|=ans[j];
		puts(bj?"YES":"NO");
	}
	return 0;
}