「bzoj4713」迷失的字符串 - bitset(未AC) | Bill Yang's Blog

「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的原因。


代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
#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;
}
0%