[Luogu P4600] [HEOI2012] [BZOJ 2746]旅行问题

最新推荐文章于 2021-07-26 17:54:12 发布

LPA20020220

最新推荐文章于 2021-07-26 17:54:12 发布

阅读量255

点赞数 1

分类专栏： LCA 字符串 AC自动机自动机文章标签： LCA 字符串 AC自动机 Trie 自动机

本文链接：https://blog.csdn.net/LPA20020220/article/details/81079357

版权

字符串同时被 3 个专栏收录

36 篇文章 1 订阅

订阅专栏

LCA

14 篇文章 0 订阅

订阅专栏

AC自动机

10 篇文章 0 订阅

订阅专栏

题目描述

yz 是 Z 国的领导人，他规定每个地区的名字只能为 $26$ 个小写拉丁字母的一个。由于地区数有可能超过 26 个，便产生了一个问题，如何辨别名字相同的地区？于是 yz 规定，一个地区的描述必须包含它的所有上级，且上级按次序排列。于是，一个地区的描述是一个字符串。比如说，一个地区的名字为 $c$ ，它的上级为 $b$ ， $b$ 的上级为 $a$ ， $a$ 没有上级，那么这个地区就描述为 $abc$ 。显然，这个描述同时包含了 $c$ 的上级 $b$ 和 $b$ 的上级 $a$ 的描述，分别为 $ab$ 和 $a$ 。

值得注意的是，每个地区最多有一个上级，同一上级的地区之间名字不同，没有上级的地区之间名字不同。

现在，yz 对外公布了 $n$ 个地区的描述，这些描述中包含了 Z 国所有地区的描述，并让你处理来访者的旅行问题。

现有 $m$ 对人访问这个国家，对于每对人，第一个人喜欢第 $i$ 个描述中的第 $j$ 个地区，设这个地区描述为 $s_1$ ，第二个人喜欢第 $k$ 个描述中的第 $l$ 个地区，设这个地区描述为 $s_2$ 。他们为了统一行程，决定访问描述为 $s$ 的地区（显然他们只关心地区的名字，并非是地区本身），设 $s$ 的长度为 $t$ ， $s$ 需要满足以下条件：

1： $t\leq j$ , $t\leq l$ ;

1： $s[1\cdots t] = s1[j-t+1\cdots j]$ , $s[1\cdots t] = s2[l-t+1\cdots l]$ ；（即 $s$ 为 $s_1$ 中 $1$ 到 $k$ 位与 $s_2$ 中 $1$ 到 $l$ 位的公共后缀）

2： $t$ 最大化。为了不使输出过大，你只需把这个字符串按照如下生成的 $26$ 进制数转成 $10$ 进制后 $mod\ 1000000007$ 后输出:

a->0

b->1

z->25

比如地区 cab 被编码成 $2\times26^2 + 0\times26^1 + 1\times26^0 = 1353$ 。

输入输出格式

输入格式：

第一行给定一个整数 $n$

第 $2\cdots n+1$ 行:每 $i+1$ 行给定一个字符串 $a[i]$ ,表示第 $i$ 个描述。

接下来一行一个整数 $m$

接下来 $m$ 行:每行给定四个整数 $i,j,k,l$ ，字母含义与题目描述一致。

输出格式：

共 $m$ 行，每行一个整数，表示答案字符串的编码。

输入输出样例

输入样例#1：

2
aabb
babb
2
1 3 2 3
1 4 2 4

输出样例#1：

1
1

说明

【样例说明】

询问 1 中的公共后有 ab 和 b，但是没有 ab 这个地区，只有 b 地区，所以只能选择 b 这个地区；

询问 2 中的公共后有 abb、bb 和 b，但是没有 abb 和 bb 这两个地区，只有 b 地区，所以只能选择 b 这个地区。

【数据范围】

设这个国家地区总数数为 tot（注意：输入的字符串总长度可能超过 tot！）

对于 30%的数据，满足 $tot，m，n\leq 100$ ；

对于 50%的数据，满足 $tot，m，n\leq 1000$ ；

对于 80%的数据，满足 $tot，m，n\leq 100000$ ；

对于 100%的数据，满足 $tot，m，n\leq 1000000$ ；

保证输入文件不超过 20MB。

解题分析

真是服了出题人，这题目描述…

题意：给你 $n$ 个串， $m$ 次询问，每次询问串 $a_i$ 的前 $j$ 位和串 $a_k$ 的前 $l$ 位的最长公共后缀，并且其满足是某个串前缀的最长长度。

思路还是有点神的：某个串前缀一定是在我们构出的 $Trie$ 树上，而前缀的后缀可以通过跳 $fail$ 链得到。所以我们答案即为 $a_i[j]$ 和 $a_k[l]$ 在 $Trie$ 树上的 $LCA$ 处的值。

那么建出 $Trie$ 树跑一边即可。

PS：博主用的倍增 $LCA$ 会被卡空间，大家最好用树剖 $LCA$ …不过即使这样洛谷也可以过（~~数据太水~~）。

代码如下：

#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <cstdlib>
#include <cctype>
#include <queue>
#define R register
#define IN inline
#define gc getchar()
#define W while
#define ll long long
#define MX 1000050
#define MOD 1000000007
template <class T>
IN void in(T &x)
{
    x = 0; R char c = gc;
    W (!isdigit(c)) c = gc;
    W (isdigit(c))
    x = (x << 1) + (x << 3) + c - 48, c = gc;
}
ll hs[MX];
struct Node
{
    int to[26];
}t[MX];
int arr, num, q, cnt;
int lent[MX], pos[MX << 1], dep[MX << 1], up[MX << 1][22];
char buf[MX];
std::queue <int> que;
namespace AC
{
    void insert(const int &ord)
    {
        int l = std::strlen(buf + 1);
        R int now = 0, val;
        lent[ord] = l + lent[ord - 1];
        for (R int i = 1; i <= l; ++i)
        {
            val = buf[i] - 'a';
            if(!t[now].to[val])
            t[now].to[val] = ++arr,
            hs[t[now].to[val]] = (hs[now] * 26 + val) % MOD;
            pos[++cnt] = t[now].to[val];
            now = t[now].to[val];
        }
    }
    void getfail()
    {
        R int now, tar;
        for (R int i = 0; i < 26; ++i) if(t[0].to[i])
        {
            que.push(t[0].to[i]);
            up[t[0].to[i]][0] = 0;
            dep[t[0].to[i]] = 1;
        }
        W (!que.empty())
        {
            now = que.front(); que.pop();
            for (R int i = 0; i < 26; ++i)
            {
                if(!t[now].to[i])//在这里倍增的是fail链， 所以依然要一直保存最近的可转移的位置。
                {
                    t[now].to[i] = t[up[now][0]].to[i];
                    continue;
                }
                tar = t[now].to[i];
                up[tar][0] = t[up[now][0]].to[i];
                dep[tar] = dep[up[tar][0]] + 1;
                for (R int i = 1; i <= 21; ++i)
                {
                    if(up[tar][i - 1])
                    up[tar][i] = up[up[tar][i - 1]][i - 1];
                    else break;
                }
                que.push(tar);
            }
        }
    }
    IN int lca(R int x, R int y)
    {
        if(dep[x] < dep[y]) std::swap(x, y);
        int del = dep[x] - dep[y];
        for (R int i = 0; i <= 21; ++i)
        {
            if(del & (1 << i)) x = up[x][i], del -= (1 << i);
            if(!del) break;
        }
        if(x == y) return x;
        for (R int i = 20; ~i; --i)
        {
            if(up[x][i] != up[y][i])
            x = up[x][i], y = up[y][i];
        }
        return up[x][0];
    }
}
int main(void)
{
    in(num); R int x, xl, y, yl, p1, p2;
    for (R int i = 1; i <= num; ++i)
    {
        scanf("%s", buf + 1);
        AC::insert(i);
    } in(q);
    AC::getfail();
    W (q--)
    {
        in(x), in(xl), in(y), in(yl);
        p1 = pos[lent[x - 1] + xl], p2 = pos[lent[y - 1] + yl];
        printf("%lld\n", hs[AC::lca(p1, p2)]);
    }
}