P2922 [USACO08DEC]秘密消息Secret Message

题目描述

Bessie is leading the cows in an attempt to escape! To do this, the cows are sending secret binary messages to each other.

Ever the clever counterspy, Farmer John has intercepted the first b_i (1 <= b_i <= 10,000) bits of each of M (1 <= M <= 50,000) of these secret binary messages.

He has compiled a list of N (1 <= N <= 50,000) partial codewords that he thinks the cows are using. Sadly, he only knows the first c_j (1 <= c_j <= 10,000) bits of codeword j.

For each codeword j, he wants to know how many of the intercepted messages match that codeword (i.e., for codeword j, how many times does a message and the codeword have the same initial bits). Your job is to compute this number.

The total number of bits in the input (i.e., the sum of the b_i and the c_j) will not exceed 500,000.

Memory Limit: 32MB

POINTS: 270

贝茜正在领导奶牛们逃跑．为了联络，奶牛们互相发送秘密信息．

信息是二进制的，共有M(1≤M≤50000)条．反间谍能力很强的约翰已经部分拦截了这些信息，知道了第i条二进制信息的前bi(l《bi≤10000)位．他同时知道，奶牛使用N(1≤N≤50000)条密码．但是，他仅仅了解第J条密码的前cj(1≤cj≤10000)位．

对于每条密码J，他想知道有多少截得的信息能够和它匹配．也就是说，有多少信息和这条密码有着相同的前缀．当然，这个前缀长度必须等于密码和那条信息长度的较小者．

在输入文件中，位的总数（即∑Bi+∑Ci）不会超过500000.

输入输出格式

输入格式：

• Line 1: Two integers: M and N

• Lines 2..M+1: Line i+1 describes intercepted code i with an integer b_i followed by b_i space-separated 0's and 1's

• Lines M+2..M+N+1: Line M+j+1 describes codeword j with an integer c_j followed by c_j space-separated 0's and 1's

输出格式：

• Lines 1..M: Line j: The number of messages that the jth codeword could match.

输入输出样例

4 5
3 0 1 0
1 1
3 1 0 0
3 1 1 0
1 0
1 1
2 0 1
5 0 1 0 0 1
2 1 1

1
3
1
1
2

说明

Four messages; five codewords.

The intercepted messages start with 010, 1, 100, and 110.

The possible codewords start with 0, 1, 01, 01001, and 11.

0 matches only 010: 1 match

1 matches 1, 100, and 110: 3 matches

01 matches only 010: 1 match

01001 matches 010: 1 match

11 matches 1 and 110: 2 matches

典型的字典树统计题。

一边建树，一边在这串数列中走过的路径中的sum+1，sum代表有sum个单词经过这个节点。

end代表有end个单词在这个节点终结。

然后读入待查询的信息，有两种情况：一是该信息全部走完，二是再往下走没有与该信息相符的节点了。

第一种情况，当前的答案要减去当前节点的end值再加上当前节点的sum值（想一想，为什么）。

第二种情况，直接输出答案。

显然比待查询的信息长的信息，如果与待查询信息有相同前缀的话，一定会经过待查询信息终结的节点。

如果待查询信息无法终结，说明没有比该信息长且前缀是该信息的信息，所以不能加上sum。

如果信息终结，那么当前节点的sum值所包含的信息一定与其有相同前缀，但sum所包含的信息有可能刚好在该节点终结，

所以要减去end值。

 1 /*
 2     典型字典树统计
 3     把字母换成01串
 4 */
 5 #include <ctype.h>
 6 #include <cstdio>
 7
 8 const int MAXN=500010;
 9
10 int n,m,tot;
11
12 int t[MAXN][3],p[MAXN],sum[MAXN],end[MAXN];
13
14 inline void read(int&x) {
15     int f=1;register char c=getchar();
16     for(x=0;!isdigit(c);c=='-'&&(f=-1),c=getchar());
17     for(;isdigit(c);x=x*10+c-48,c=getchar());
18     x=x*f;
19 }
20
21 inline void build() {
22     int now=0;
23     for(int i=1;i<=p[0];++i) {
24         if(!t[now][p[i]]) t[now][p[i]]=++tot;
25         now=t[now][p[i]];
26         ++sum[now];
27     }
28     ++end[now];
29     return;
30 }
31
32 inline void find() {
33     int ans=0,now=0;
34     bool f=true;
35     for(int i=1;i<=p[0];++i) {
36         if(t[now][p[i]]) now=t[now][p[i]];
37         else {
38             f=false;
39             break;
40         }
41         ans+=end[now];
42     }
43     if(!f) printf("%d\n",ans);
44     else printf("%d\n",ans+sum[now]-end[now]);
45     return;
46 }
47
48 int hh() {
49     read(m);read(n);
50     for(int i=1;i<=m;++i)  {
51         read(p[0]);
52         for(int i=1;i<=p[0];++i) read(p[i]);
53         build();
54     }
55     for(int i=1;i<=n;++i) {
56         read(p[0]);
57         for(int i=1;i<=p[0];++i) read(p[i]);
58         find();
59     }
60     return 0;
61 }
62
63 int sb=hh();
64 int main() {;}

代码