【后缀数组】【SCOI2012】喵星球上的点名

本文介绍了一种用于喵星球上点名的算法实现,通过后缀数组等数据结构,高效统计点名过程中喵星人响应的情况。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

Description
a180285幸运地被选做了地球到喵星球的留学生。他发现喵星人在上课前的点名现象非常有趣。   假设课堂上有N个喵星人,每个喵星人的名字由姓和名构成。喵星球上的老师会选择M个串来点名,每次读出一个串的时候,如果这个串是一个喵星人的姓或名的子串,那么这个喵星人就必须答到。 然而,由于喵星人的字码过于古怪,以至于不能用ASCII码来表示。为了方便描述,a180285决定用数串来表示喵星人的名字。
现在你能帮助a180285统计每次点名的时候有多少喵星人答到,以及M次点名结束后每个喵星人答到多少次吗?  
Input
 
现在定义喵星球上的字符串给定方法:
先给出一个正整数L,表示字符串的长度,接下来L个整数表示字符串的每个字符。
输入的第一行是两个整数N和M。
接下来有N行,每行包含第i 个喵星人的姓和名两个串。姓和名都是标准的喵星球上的
字符串。
接下来有M行,每行包含一个喵星球上的字符串,表示老师点名的串。
Output
 
对于每个老师点名的串输出有多少个喵星人应该答到。
然后在最后一行输出每个喵星人被点到多少次。
Sample Input
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

Sample Output
2
1
0
1 2
【提示】
事实上样例给出的数据如果翻译成地球上的语言可以这样来看
2 3
izayoi sakuya
orihara izaya
izay
hara
raiz
HINT
【数据范围】

 对于30%的数据,保证:
1<=N,M<=1000,喵星人的名字总长不超过4000,点名串的总长不超过2000。
对于100%的数据,保证:
1<=N<=20000,1<=M<=50000,喵星人的名字总长和点名串的总长分别不超过100000,保证喵星人的字符串中作为字符存在的数不超过10000。
后缀数组。
此题水题,过了省选就过了。。。
Accode:

#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <string>
#include <cstring>
#include <bitset>

const int maxN = 400010;
bool marked[maxN];
int wa[maxN], wb[maxN], ws[maxN], wv[maxN];
int r[maxN], rank[maxN], height[maxN], sa[maxN];
int name_l[2][maxN], sub_l[maxN];
int name_p[2][maxN], sub_p[maxN];
int name_c[maxN], sub_c[maxN], stack[maxN];
int n, m, N, top;

inline bool cmp(int *r, int a, int b, int len)
{return r[a] == r[b] && r[a + len] == r[b + len];}

void da(int *r, int *sa, int n, int m)
{
    int *x = wa, *y = wb;
    for (int i = 0; i < m; ++i) ws[i] = 0;
    for (int i = 0; i < n; ++i) ++ws[x[i] = r[i]];
    for (int i = 1; i < m; ++i) ws[i] += ws[i - 1];
    for (int i = n - 1; i > -1; --i) sa[--ws[x[i]]] = i;
    for (int j = 1, p = 1; p < n; j <<= 1, m = p)
    {
        p = 0;
        for (int i = n - j; i < n; ++i) y[p++] = i;
        for (int i = 0; i < n; ++i)
            if (sa[i] - j > -1) y[p++] = sa[i] - j;
        for (int i = 0; i < n; ++i) wv[i] = x[y[i]];
        for (int i = 0; i < m; ++i) ws[i] = 0;
        for (int i = 0; i < n; ++i) ++ws[wv[i]];
        for (int i = 1; i < m; ++i) ws[i] += ws[i - 1];
        for (int i = n - 1; i > -1; --i) sa[--ws[wv[i]]] = y[i];
        std::swap(x, y); x[sa[0]] = 0; p = 1;
        for (int i = 1; i < n; ++i)
            x[sa[i]] = cmp(y, sa[i - 1], sa[i], j) ? p - 1 : p++;
    }
    return;
}

void calheight(int *r, int *sa, int n)
{
    int k = 0;
    for (int i = 1; i < n + 1; ++i) rank[sa[i]] = i;
    for (int i = 0; i < n; height[rank[i++]] = k)
    {
        int j = sa[rank[i] - 1]; if (k) --k;
        while (r[i + k] && r[i + k] == r[j + k]) ++k;
    }
    return;
}

inline int getint()
{
    int res = 0; char tmp; bool sgn = 1;
    do tmp = getchar();
    while (!isdigit(tmp) && tmp - '-');
    if (tmp == '-') {sgn = 0; tmp = getchar();}
    do res = (res << 3) + (res << 1) + tmp - '0';
    while (isdigit(tmp = getchar()));
    return sgn ? res : -res;
}

inline int Find(int x)
{
    if (x < name_p[1][0]) return 0;
    int L = 1, R = n - 1;
    while (L < R + 1)
    {
        int Mid = L + R >> 1;
        if (x < name_p[1][Mid] &&
            x > name_p[1][Mid - 1])
            return Mid;
        if (x > name_p[1][Mid]) L = Mid + 1;
        else R = Mid - 1;
    }
    return -1;
}

int main()
{
    freopen("name.in", "r", stdin);
    freopen("name.out", "w", stdout);
    n = getint(); m = getint(); N = 0;
    for (int i = 0; i < n; ++i)
    {
        name_l[0][i] = getint();
        for (int j = 0; j < name_l[0][i]; ++j)
            r[N++] = getint() + 1;
        r[name_p[0][i] = (N++)] = 0;
        
        name_l[1][i] = getint();
        for (int j = 0; j < name_l[1][i]; ++j)
            r[N++] = getint() + 1;
        r[name_p[1][i] = (N++)] = 0;
    }
    for (int i = 0; i < m; ++i)
    {
        sub_l[i] = getint();
        for (int j = 0; j < sub_l[i]; ++j)
            r[N++] = getint() + 1;
        r[sub_p[i] = (N++)] = 0;
    }
    da(r, sa, N--, 10010);
    calheight(r, sa, N);
    for (int i = 0; i < m; ++i)
    {
        int pos = rank[sub_p[i] - sub_l[i]];
        top = 0;
        for (int j = pos; j; --j)
        {
            if (height[j] < sub_l[i]) break;
            int tmp = Find(sa[j - 1]);
            if (tmp + 1)
            {
                if (!marked[tmp])
                    ++name_c[stack[top++] = tmp], ++sub_c[i];
                marked[tmp] = 1;
            }
        }
        for (int j = pos + 1; j < N + 1; ++j)
        {
            if (height[j] < sub_l[i]) break;
            int tmp = Find(sa[j]);
            if (tmp + 1)
            {
                if (!marked[tmp])
                    ++name_c[stack[top++] = tmp], ++sub_c[i];
                marked[tmp] = 1;
            }
        }
        for (int j = 0; j < top; ++j)
            marked[stack[j]] = 0;
    }
    for (int i = 0; i < m; ++i)
        printf("%d\n", sub_c[i]);
    for (int i = 0; i < n; ++i)
        printf("%d ", name_c[i]);
    printf("\n");
    return 0;
}

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值