[BZOJ2754][SCOI2012]喵星球上的点名
[BZOJ2754][SCOI2012]喵星球上的点名
试题描述
a180285幸运地被选做了地球到喵星球的留学生。他发现喵星人在上课前的点名现象非常有趣。 假设课堂上有N个喵星人，每个喵星人的名字由姓和名构成。喵星球上的老师会选择M个串来点名，每次读出一个串的时候，如果这个串是一个喵星人的姓或名的子串，那么这个喵星人就必须答到。 然而，由于喵星人的字码过于古怪，以至于不能用ASCII码来表示。为了方便描述，a180285决定用数串来表示喵星人的名字。
现在你能帮助a180285统计每次点名的时候有多少喵星人答到，以及M次点名结束后每个喵星人答到多少次吗？
输入
现在定义喵星球上的字符串给定方法：
先给出一个正整数L，表示字符串的长度，接下来L个整数表示字符串的每个字符。
输入的第一行是两个整数N和M。
接下来有N行，每行包含第i 个喵星人的姓和名两个串。姓和名都是标准的喵星球上的字符串。
接下来有M行，每行包含一个喵星球上的字符串，表示老师点名的串。
输出
对于每个老师点名的串输出有多少个喵星人应该答到。
然后在最后一行输出每个喵星人被点到多少次。
输入示例
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
输出示例
2 1 0 1 2
数据规模及约定
对于30%的数据，保证：
1<=N,M<=1000，喵星人的名字总长不超过4000，点名串的总长不超过2000。
对于100%的数据，保证：
1<=N<=20000，1<=M<=50000，喵星人的名字总长和点名串的总长分别不超过100000，保证喵星人的字符串中作为字符存在的数不超过10000。
题解
求后缀数组，对于每一次点名，点到的对应了一个区间，对于第一问就是问区间中有多少个不同的数，用个主席树就好了；对于第二问不会做，暴力骗过。。。
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cctype>
#include <algorithm>
using namespace std;
int read() {
int x = 0, f = 1; char c = getchar();
while(!isdigit(c)){ if(c == '-') f = -1; c = getchar(); }
while(isdigit(c)){ x = x * 10 + c - '0'; c = getchar(); }
return x * f;
}
#define maxn 300010
#define maxN 20010
#define maxM 50010
#define maxlog 19
#define maxnode 6000010
int n, m, Str[maxn], tag[maxn], lst[maxN], pre[maxn], st[maxM], Len[maxM], len;
int rank[maxn], height[maxn], sa[maxn], Ws[maxn];
bool cmp(int* a, int p1, int p2, int l) {
if(p1 + l > len && p2 + l > len) return a[p1] == a[p2];
if(p1 + l > len || p2 + l > len) return 0;
return a[p1] == a[p2] && a[p1+l] == a[p2+l];
}
void ssort() {
int *x = rank, *y = height;
int m = 0;
for(int i = 1; i <= len; i++) Ws[x[i] = Str[i]]++, m = max(x[i], m);
for(int i = 1; i <= m; i++) Ws[i] += Ws[i-1];
for(int i = len; i; i--) sa[Ws[x[i]]--] = i;
for(int j = 1, pos = 0; pos < len; j <<= 1, m = pos) {
pos = 0;
for(int i = len - j + 1; i <= len; i++) y[++pos] = i;
for(int i = 1; i <= len; i++) if(sa[i] > j) y[++pos] = sa[i] - j;
for(int i = 1; i <= m; i++) Ws[i] = 0;
for(int i = 1; i <= len; i++) Ws[x[i]]++;
for(int i = 1; i <= m; i++) Ws[i] += Ws[i-1];
for(int i = len; i; i--) sa[Ws[x[y[i]]]--] = y[i];
swap(x, y); pos = 1; x[sa[1]] = 1;
for(int i = 2; i <= len; i++) x[sa[i]] = cmp(y, sa[i], sa[i-1], j) ? pos : ++pos;
}
return ;
}
void calch() {
for(int i = 1; i <= len; i++) rank[sa[i]] = i;
for(int i = 1, j, k = 0; i <= len; height[rank[i++]] = k)
for(k ? k-- : 0, j = sa[rank[i]-1]; Str[i+k] == Str[j+k]; k++);
return ;
}
int Log[maxn], mnh[maxlog][maxn];
void rmq_init() {
Log[1] = 0;
for(int i = 2; i <= len; i++) Log[i] = Log[i>>1] + 1;
for(int i = 1; i <= len; i++) mnh[0][i] = height[i];
for(int j = 1; (1 << j) <= len; j++)
for(int i = 1; i + (1 << j) - 1 <= len; i++)
mnh[j][i] = min(mnh[j-1][i], mnh[j-1][i+(1<<j-1)]);
return ;
}
int query(int l, int r) {
l++;
if(l > r) return len - sa[r] + 1;
int t = Log[r-l+1];
return min(mnh[t][l], mnh[t][r-(1<<t)+1]);
}
int ToT, rt[maxn], sumv[maxnode], lc[maxnode], rc[maxnode];
void update(int& y, int x, int l, int r, int p) {
sumv[y = ++ToT] = sumv[x] + 1;
if(l == r) return ;
int mid = l + r >> 1; lc[y] = lc[x]; rc[y] = rc[x];
if(p <= mid) update(lc[y], lc[x], l, mid, p);
else update(rc[y], rc[x], mid + 1, r, p);
return ;
}
int que(int o, int l, int r, int up) {
if(r <= up) return sumv[o];
int mid = l + r >> 1, ans = que(lc[o], l, mid, up);
if(up > mid) ans += que(rc[o], mid + 1, r, up);
return ans;
}
int ask(int l, int r) {
return que(rt[r], 0, len, l - 1) - que(rt[l-1], 0, len, l - 1);
}
int vis[maxM], hhh[maxN];
int main() {
n = read(); m = read();
for(int i = 1; i <= n; i++) {
int L = read();
while(L--) Str[++len] = read(), tag[len] = i;
Str[++len] = 10001;
L = read();
while(L--) Str[++len] = read(), tag[len] = i;
Str[++len] = 10001;
}
for(int i = 1; i <= m; i++) {
int L = read();
st[i] = len + 1; Len[i] = L;
while(L--) Str[++len] = read();
Str[++len] = 10001;
}
ssort();
calch();
rmq_init();
/*for(int i = 1; i <= len; i++) printf("%d%c", Str[i], i < len ? ' ' : '\n');
for(int i = 1; i <= len; i++) printf("%d%c", sa[i], i < len ? ' ' : '\n');
for(int i = 1; i <= len; i++) printf("%d%c", tag[sa[i]], i < len ? ' ' : '\n');
for(int i = 1; i <= len; i++) printf("%d%c", height[i], i < len ? ' ' : '\n'); // */
for(int i = 1; i <= len; i++) pre[i] = lst[tag[sa[i]]], lst[tag[sa[i]]] = i;
for(int i = 1; i <= len; i++) update(rt[i], rt[i-1], 0, len, pre[i]);
for(int i = 1; i <= m; i++) {
int now = rank[st[i]], l = 1, r = now, L, R;
while(l < r) {
int mid = l + r >> 1;
if(query(mid, now) < Len[i]) l = mid + 1; else r = mid;
}
L = l;
l = now; r = len + 1;
while(r - l > 1) {
int mid = l + r >> 1;
if(query(now, mid) < Len[i]) r = mid; else l = mid;
}
R = l;
// printf("%d [%d, %d]\n", now, L, R);
printf("%d\n", ask(L, R) - 1);
for(int j = L; j <= R; j++) if(tag[sa[j]] && vis[tag[sa[j]]] != i) vis[tag[sa[j]]] = i, hhh[tag[sa[j]]]++;
}
for(int i = 1; i <= n; i++) printf("%d%c", hhh[i], i < n ? ' ' : '\n');
return 0;
}
真 TM 调死我了 = =
