洛谷 P2336 [SCOI2012]喵星球上的点名

题目描述

a180285 幸运地被选做了地球到喵星球的留学生。他发现喵星人在上课前的点名现象非常有趣。

假设课堂上有 N 个喵星人，每个喵星人的名字由姓和名构成。喵星球上的老师会选择M 个串来点名，每次读出一个串的时候，如果这个串是一个喵星人的姓或名的子串，那么这个喵星人就必须答到。

然而，由于喵星人的字码过于古怪，以至于不能用 ASCII 码来表示。为了方便描述，a180285 决定用数串来表示喵星人的名字。

现在你能帮助 a180285 统计每次点名的时候有多少喵星人答到，以及 M 次点名结束后每个喵星人答到多少次吗？

输入输出格式

输入格式：

 

现在定义喵星球上的字符串给定方法：

先给出一个正整数 L ，表示字符串的长度，接下来L个整数表示字符串的每个字符。

输入的第一行是两个整数 N 和 M。

接下来有 N 行， 每行包含第 i 个喵星人的姓和名两个串。 姓和名都是标准的喵星球上的字符串。

接下来有 M 行，每行包含一个喵星球上的字符串，表示老师点名的串。

 

输出格式：

 

对于每个老师点名的串输出有多少个喵星人应该答到。

然后在最后一行输出每个喵星人被点到多少次。

 

输入输出样例

输入样例#1：

2 3
6 8 25 0 24 14 8 6 18 0 10 20 24 0
7 14 17 8 7 0 17 0 5 8 25 0 24 0
4 8 25 0 24
4 7 0 17 0
4 17 0 8 25

输出样例#1：

2
1
0
1 2

说明

事实上样例给出的数据如果翻译成地球上的语言可以这样来看

2 3

izayoi sakuya

orihara izaya

izay hara raiz

对于 30%的数据，保证：

1<=N,M<=1000,喵星人的名字总长不超过 4000，点名串的总长不超过 2000，

对于 100%的数据，保证：

1<=N<=20000 ，1<=M<=50000. 喵星人的名字总长和点名串的总长分别不超过100000,保证喵星人的字符串中作为字符存在的数不超过 10000

题目大意：多个模板串多个匹配串，求每个模板串被匹配了多少次，每个匹配串匹配了多少模板，字符集很大

要命题，调了一下午

AC自动机的最坏复杂度是n^2，然而这题没有卡，可以过

后缀数组+ST表+二分+莫队+差分

首先把这么一大堆东西串起来，像这样

izayoi#sakuya#orihara#izaya#izay#hara#raiz#

具体实现的话所有字符+1，0充当分隔符就好了

然后对每一个名字所占据的位置记录一下颜色，像这样

izayoi#sakuya#orihara#izaya#izay#hara#raiz#
1111111111111122222222222222000000000000000

记录一下每一个点名的起点和长度

然后把sa和height求出来

先搞定每个匹配串匹配多少模板，把height的ST表搞出来，二分一下height>点名长度的区间，这个区间的中间位置大概在rank[点名起点]，但也可能是个空区间，需要特判一下

然后就问题变成了区间颜色个数，莫队模板题

然后处理每个模板串被多少匹配串匹配过

做一个差分，每次在莫队添加和删除元素的时候，如果把一种颜色删光了，给这种颜色的出现次数减去剩下的询问数量，如果一种颜色新出现了，加上剩下的询问数量

#include <iostream>
#include <cstdlib>
#include <cstdio>
#include <algorithm>
#include <string>
#include <cstring>
#include <cmath>
#include <map>
#include <stack>
#include <set>
#include <vector>
#include <queue>
#include <time.h>
#define eps 1e-7
#define INF 0x3f3f3f3f
#define MOD 1000000007
#define rep0(j,n) for(int j=0;j<n;++j)
#define rep1(j,n) for(int j=1;j<=n;++j)
#define pb push_back
#define mp make_pair
#define set0(n) memset(n,0,sizeof(n))
#define ll long long
#define ull unsigned long long
#define iter(i,v) for(edge *i=head[v];i;i=i->nxt)
#define max(a,b) (((a)>(b))?(a):(b))
#define min(a,b) (((a)<(b))?(a):(b))
#define print_runtime printf("Running time:%.3lfs
",double(clock())/1000.0)
#define TO(j) printf(#j": %d
",j);
//#define OJ
using namespace std;
const int MAXINT = 401000;
const int MAXNODE = 100010;
const int MAXEDGE = 2 * MAXNODE;
char BUF, *buf;
int read() {
    char c = getchar(); int f = 1, x = 0;
    while (!isdigit(c)) { if (c == '-') f = -1; c = getchar(); }
    while (isdigit(c)) { x = x * 10 + c - '0'; c = getchar(); }
    return f * x;
}
char get_ch() {
    char c = getchar();
    while (!isalpha(c)) c = getchar();
    return c;
}
//------------------- Head Files ----------------------//
int cnt_qr, sa[MAXINT], rk[MAXINT], trk[MAXINT], height[MAXINT], tsa[MAXINT], tstr[MAXINT], cnt[MAXINT], len, s[MAXINT], n, m, color[MAXINT];
int st[18][MAXINT], maxp[MAXINT*2], ans=0, cnt_c[50010], cnt_miao[20010], rp;
pair<int, int> qr[50010];
struct query {
    int l, r, id;
    query() {}
    query(int _l, int _r, int _id) : l(_l), r(_r), id(_id) {}
} mo[MAXINT];
bool operator < (const query &a, const query &b) {
    return (a.l / 447 == b.l / 447) ? a.r < b.r : (a.l / 447 < b.l / 447);
}
int cmp(int *a, int p1, int p2, int l) {
    return a[p1] == a[p2] && a[p1 + l] == a[p2 + l];
}
void get_st() {
    rep0(i, 18) {
        for (int j = (1 << i); j < (1 << (i + 1)); j++) maxp[j] = i;
    }
    rep0(i, len) st[0][i] = height[i];
    rep1(i, 17) rep0(j, len - (1 << i) + 1) st[i][j] = min(st[i - 1][j], st[i - 1][j + (1 << (i - 1))]);
}
int gt(int l, int r) { //[l,r]
    int pw = maxp[r - l + 1];
    return min(st[pw][l], st[pw][r - (1 << pw) + 1]);
}
void get_sa() {
    int i, j, k, sz = MAXINT;
    for (i = 0; i < sz; i++) cnt[i] = 0;
    for (i = 0; i < len; i++) cnt[trk[i] = s[i]]++;
    for (i = 1; i < sz; i++) cnt[i] += cnt[i - 1];
    for (i = len - 1; i >= 0; i--) sa[--cnt[s[i]]] = i;
    for (j = 1, k = 1; k < len; j <<= 1, sz = k) {
        for (k = 0, i = len - j; i < len; i++) tsa[k++] = i;
        for (i = 0; i < len; i++) if (sa[i] >= j) tsa[k++] = sa[i] - j;
        for (i = 0; i < len; i++) tstr[i] = trk[tsa[i]];

        for (i = 0; i < sz; i++) cnt[i] = 0;
        for (i = 0; i < len; i++) cnt[tstr[i]]++;
        for (i = 1; i < sz; i++) cnt[i] += cnt[i - 1];
        for (i = len - 1; i >= 0; i--) sa[--cnt[tstr[i]]] = tsa[i];
        for (swap(tsa, trk), k = 1, trk[sa[0]] = 0, i = 1; i < len; i++) {
            trk[sa[i]] = cmp(tsa, sa[i], sa[i - 1], j) ? k - 1 : k++;
        }
    }
}
void get_height() {
    int i, j, k;
    for (i = 0; i < len; i++) rk[sa[i]] = i;
    for (i = 0, k = 0; i < len - 1; height[rk[i++]] = k) {
        for (k ? k-- : 0, j = sa[rk[i] - 1]; s[i + k] == s[j + k]; k++);
    }
}
query get_query(int id) {
    int pos = rk[qr[id].first], l = qr[id].second;
    query ret; ret.id = id;
    if (pos == len - 1) pos = pos + 1;
    else {
        if (height[pos] < l) pos = pos + 1;
    }
    if (height[pos] < l) return query(-1, -1, -1);
    int mid, lb = pos, rb = len;
    while (rb - lb > 1) {
        mid = (lb + rb) / 2;
        if (gt(pos, mid) < l) rb = mid; else lb = mid;
    }
    ret.r = lb;
    lb = 0; rb = pos;
    while (rb - lb > 1) {
        mid = (lb + rb) / 2;
        if (gt(mid, pos) < l) lb = mid; else rb = mid;
    }
    ret.l = rb - 1;
    return ret;
}
void get_input();
void work();
void append() {
    int l = read();
    rep0(i, l) {
        int w = read();
        s[len++] = w + 1;
    }
    s[len++] = 0;
}
void add(int v) {
    cnt[v]++;
    if (v != 0 && cnt[v] == 1) {
        ans++;
        cnt_miao[v] += cnt_qr - rp;
    }
}
void del(int v) {
    cnt[v]--;
    if (v!=0 && cnt[v] == 0) {
        ans--;
        cnt_miao[v] -= cnt_qr - rp;
    };
}
int main() {
    get_input();
    work();
    return 0;
}
void work() {
    get_sa();
    get_height();
    get_st();
    rep0(i, m) {
        mo[cnt_qr] = get_query(i);
        if (mo[cnt_qr].id != -1) cnt_qr++;
    }
    sort(mo, mo + cnt_qr);
    memset(cnt, 0, sizeof(cnt));
    int l = 0, r = -1;
    for (rp = 0; rp < cnt_qr; rp++) {
        for (; l < mo[rp].l; del(color[sa[l]]), l++);
        for (; l > mo[rp].l; l--, add(color[sa[l]]));
        for (; r > mo[rp].r; del(color[sa[r]]), r--);
        for (; r < mo[rp].r; r++, add(color[sa[r]]));
        cnt_c[mo[rp].id] = ans; l = mo[rp].l; r = mo[rp].r;
    }
    rep0(i, m) printf("%d
", cnt_c[i]);
    rep1(i, n) printf("%d ", cnt_miao[i]);
    putchar('
');
}
void get_input() {
    n = read(); m = read();
    int l;
    rep1(i, n) {
        int p = len;
        append(); append();
        for (int j = p; j < len; j++) color[j] = i;
    }
    rep0(i, m) {
        int p = len;
        append();
        for (int j = p; j < len; j++) color[j] = 0;
        qr[i] = make_pair(p, len - p - 1);
    }
}
/*
2 1
2 2 1 1 1
2 2 1 1 1
2 2 1
*/