洛谷 P2922 Secret Message G 题解

洛谷 P2922 Secret Message G 题解

题目

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

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

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

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

输入输出

输入
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

解析

首先要理解题目：

解释一下样例：
有4条信息，分别是：
0 1 0
1
1 0 0
1 1 0
有5个暗号，分别是：
0
1
0 1
0 1 0 0 1
1 1

0所匹配的信息是0 1 0，共1条
1所匹配的信息是1，1 0 0，1 1 0，共3条
0 1所匹配的信息是0 1 0，共1条
0 1 0 0 1所匹配的信息是0 1 0，共1条
1 1所匹配的信息是1 1 0，1，共2条
所以输出1 3 1 1 2

代码

#include<bits/stdc++.h>
using namespace std;
int n,m;
int next[500005][3],is_end[500005],tot = 0,sum[500005];
void add(string s,int len){
	int p = 0;
	for(int i = 0;i<len;i++) {
		int c = s[i]-48;
		if(next[p][c] == 0){
			tot++;
			next[p][c] = tot;
		}
		p = next[p][c];
		sum[p]++;
	}
	is_end[p]++;
}

int search(string s,int len) {
	int p = 0;
	int result = 0;
	for(int i = 0;i<len;i++) {
		int c = s[i]-0x30;
		if(next[p][c] == 0){
			return result;
		}
		p = next[p][c];
		result+=is_end[p];
	}
	return result-is_end[p]+sum[p];
}

int main() {
	scanf("%d %d",&n,&m);
	for(int i = 0;i<n;i++) {
		int len; 
		scanf("%d",&len);
		string message;
		while(1){
			char c;
			c = getchar();
			if(c == '\n'){
				break;
			}else if(c!=' '){
				message+=c;
			}
		}
		add(message,len);
	}
	for(int i = 0;i<m;i++) {
		int len;
		string codeword;
		scanf("%d",&len);
		while(1){
			char c;
			c = getchar();
			if(c == '\n'){
				break;
			}else if(c!=' '){
				codeword+=c;
			}
		}
		printf("%d\n",search(codeword,len));
	}
}