说出来你可能不信,我把数组开成了char,结果检查了一辈子没检查出来。
这题乍一看,是后缀自动机。但是后缀自动机都忘完了。
于是就想了想,能不能用后缀数组。
首先,分析完题目。其实题目让求的就是和询问串的后缀相同的,而且作为P串的子串的种数相同的有多少个。
于是。
我们可以把P串都接到一块,用不同的字符隔开。
每次来一个串,我们可以通过二分,确定公共子串是当前串的所有后缀串前缀(把所有串翻转一下,就变成了找后缀的前缀,就是子串)有哪些。然后显然能使用的最长长度就是这些串的最长公共长度。而最短长度就是可以使得个数变多的所求串的前缀的长度长度+1。于是减一下就可以了。
注意,当串只有一个的时候需要特殊处理一下。
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<string>
using namespace std;
const int maxn = 200030;
char ch[maxn], ch2[maxn];
int bg[maxn], id[maxn];
struct SuffixArray {
int r[maxn];
int sa[maxn], rank[maxn], height[maxn];
int t[maxn], t2[maxn], c[maxn], n;
int m;///模板长度
void init(char* s) {
n = strlen(s);
int cnt = 'z' + 1;
for (int i = 0; i < n; i++) {
r[i] = int(s[i]);
if(s[i] == '#') r[i] = cnt ++;
}
m = cnt;
}
int cmp(int *r, int a, int b, int l) {
return r[a] == r[b] && r[a + l] == r[b + l];
}
/**
字符要先转化为正整数
待排序的字符串放在r[]数组中,从r[0]到r[n-1],长度为n,且最大值小于m。
所有的r[i]都大于0,r[n]无意义算法中置0
函数结束后,结果放在sa[]数组中(名次从1..n),从sa[1]到sa[n]。s[0]无意义
**/
void build_sa() {
int i, k, p, *x = t, *y = t2;
r[n++] = 0;
for (i = 0; i < m; i++) c[i] = 0;
for (i = 0; i < n; i++) c[x[i] = r[i]]++;
for (i = 1; i < m; i++) c[i] += c[i - 1];
for (i = n - 1; i >= 0; i--) sa[--c[x[i]]] = i;
for (k = 1, p = 1; k < n; k *= 2, m = p) {
for (p = 0, i = n - k; i < n; i++) y[p++] = i;
for (i = 0; i < n; i++) if (sa[i] >= k) y[p++] = sa[i] - k;
for (i = 0; i < m; i++) c[i] = 0;
for (i = 0; i < n; i++) c[x[y[i]]]++;
for (i = 1; i < m; i++) c[i] += c[i - 1];
for (i = n - 1; i >= 0; i--) sa[--c[x[y[i]]]] = y[i];
swap(x, y);
p = 1;
x[sa[0]] = 0;
for (i = 1; i < n; i++) x[sa[i]] = cmp(y, sa[i - 1], sa[i], k) ? p - 1 : p++;
}
n--;
}
/**
height[2..n]:height[i]保存的是lcp(sa[i],sa[i-1])
rank[0..n-1]:rank[i]保存的是原串中suffix[i]的名次
**/
void getHeight() {
int i, j, k = 0;
for (i = 1; i <= n; i++) rank[sa[i]] = i;
for (i = 0; i < n; i++) {
if (k) k--;
j = sa[rank[i] - 1];
while (r[i + k] == r[j + k]) k++;
height[rank[i]] = k;
}
}
int d[maxn][20];
///元素从1编号到n
void RMQ_init(int A[], int n) {
for (int i = 1; i <= n; i++) d[i][0] = A[i];
for (int j = 1; (1 << j) <= n; j++)
for (int i = 1; i + (1 << (j - 1)) <= n; i++)
d[i][j] = min(d[i][j - 1], d[i + (1 << (j - 1))][j - 1]);
}
int RMQ(int L, int R) {
int k = 0;
if(L > R) swap(L, R);
if(L == R) return n - sa[L];
L++;
while ((1 << (k + 1)) <= R - L + 1) k++;
return min(d[L][k], d[R - (1 << k) + 1][k]);
}
void call_fun(char* s) {
init(s);
build_sa();
getHeight();
RMQ_init(height, n);
}
int gao(char* ch2) {
int len = strlen(ch2);
int L = 1, R = n, cnt = n + 1, ret = 0;
for(int i = len - 1, j = 0; i >= 0; i --, j ++) {
int L1 = L, R1 = R;
while(L1 <= R1) {
int mid = (L1 + R1) >> 1;
if(n - sa[mid] - j <= 0 || r[sa[mid] + j] < ch2[i]) L1 = mid + 1;
else R1 = mid - 1;
}
L = L1;
L1 = L, R1 = R;
while(L1 <= R1) {
int mid = (L1 + R1) >> 1;
if(r[sa[mid] + j] == ch2[i]) L1 = mid + 1;
else R1 = mid - 1;
}
R = R1;
if(L > R) return 0;
if(R - L < cnt) ret = 1, cnt = R - L;
else if(R - L == cnt) ret ++;
}
if(L == R) {
return ret + bg[id[sa[L]]] - sa[L] - len;
}
return RMQ(L, R) - len + ret;
}
} sol;
int main() {
int T, n, m, cas = 1;
scanf("%d", &T);
while(T --) {
scanf("%d%d", &n, &m);
int l = 0;
for(int i = 0; i < n; i ++) {
scanf("%s", ch2);
int l2 = strlen(ch2);
for(int j = 0; j < l2; j ++) {
id[l] = i;
ch[l ++] = ch2[l2 - j - 1];
}
bg[i] = l;
id[l] = i;
ch[l ++] = '#';
}
ch[l] = 0;
sol.call_fun(ch);
printf("Case #%d:\n", cas ++);
for(int i = 0; i < m; i ++) {
scanf("%s", ch2);
printf("%d\n", sol.gao(ch2));
}
}
return 0;
}