洛谷 - P2412 查单词

难度:普及/提高-
标签:字符串 搜索 线段树 洛谷原创

题面

题目描述

蒟蒻 HansBug 在一本英语书里面找到了一个单词表,包含 nn 个单词(每个单词内包含大小写字母)。现在他想要找出某一段连续的单词内字典序最大的单词。

输入格式

第一行包含两个正整数 n,mn, m ,分别表示单词个数和询问个数。

接下来 nn 行每行包含一个字符串,仅包含大小写字母,长度不超过 1515 ,表示一个单词。

再接下来 mm 行每行包含两个整数 x,yx, y ,表示求从第 xx 到第 yy 个单词中字典序最大的单词。

输出格式

输出包含 mm 行,每行为一个字符串,分别依次对应前面 mm 个询问的结果。

输入输出样例

输入样例 #1

5 5
absi
hansbug
lzn
kkk
yyy
1 5
1 1
1 2
2 3
4 4

输出样例 #1

yyy
absi
hansbug
lzn
kkk

样例说明 #1

第一次操作:在 {absi, hansbug, lzn, kkk, yyy}\text{\{absi,\ hansbug,\ lzn,\ kkk,\ yyy\}}中找出字典序最大的,故为 yyy\text{yyy}
第二次操作:在 {absi}\text{\{absi\}} 中找出字典序最大的,故为 absi\text{absi}
第三次操作:在 {absi,hansbug}\text{\{absi,hansbug\}} 中找出字典序最大的,故为 hansbug\text{hansbug}
第四次操作:在 {hansbug,lzn}\text{\{hansbug,lzn\}} 中找出字典序最大的,故为 lzn\text{lzn}
第五次操作:在 {kkk}\text{\{kkk\}} 中找出字典序最大的,故为 kkk\text{kkk}

数据规模与约定

该题目单词字典序比对过程中大小写不敏感,但是输出必须输出原单词。

对于 30%30\% 的数据, n10,m10n \leq 10, m \leq 10
对于 70%70\% 的数据, n50000,m50000n \leq 50000, m \leq 50000
对于 100%100\% 的数据, n50000,m300000n \leq 50000, m \leq 300000

思路

题目大意:输出区间字典序最大的字符串,比对过程中大小写不敏感,但是输出必须输出原单词

由题可得设计出的程序需要支持区间查询的操作,进而想到使用线段树(可以参考这篇博文:线段树学习笔记)。

这里有一个坑点(踩坑记录 R52371435:C++ 中字符顺序是按照 ASCII 码来比较的,如 A < B < a\text{A < B < a} ,所以需要统一转换为大写或小写,转换后可以使用 std::map 存储字符对应关系。

之后按照正常线段树操作编写程序即可。

代码

#include <bits/stdc++.h>

using namespace std;

struct node {
    int l, r;
    string s;

    node() {
        l = r = 0;
        s = "";
    }
    node(int _l, int _r) {
        l = _l;
        r = _r;
        s = "";
    }
} tr[50005 << 2];
int n, m, x, y;
string a[50005], b[50005];
map<string, string> m1;

void pushup(int u) {
    tr[u].s = max(tr[u << 1].s, tr[u << 1 | 1].s);
}

void build(int u, int l, int r) {
    tr[u] = node(l, r);
    if (l == r) {
        tr[u].s = a[l];
        return;
    }
    int mid = l + r >> 1;
    build(u << 1, l, mid);
    build(u << 1 | 1, mid + 1, r);
    pushup(u);
}

string query(int u, int l, int r) {
    if (l <= tr[u].l && tr[u].r <= r) {
        return tr[u].s;
    }
    int mid = tr[u].l + tr[u].r >> 1;
    string s;
    if (l <= mid) s = max(s, query(u << 1, l, r));
    if (r > mid) s = max(s, query(u << 1 | 1, l, r));
    return s;
}

int main() {
    cin >> n >> m;
    for (int i = 1; i <= n; i++) {
        cin >> b[i];
        for (char c : b[i]) {
            a[i].push_back(tolower(c));
        }
        m1[a[i]] = b[i];
    }
    build(1, 1, n);
    for (int i = 0; i < m; i++) {
        cin >> x >> y;
        cout << m1[query(1, x, y)] << endl;
    }
    return 0;
}

评测记录 R52376877 - 洛谷

洛谷 - P2412 查单词
本文作者
宝硕
发布于
2021-07-04
更新于
2021-08-08
许可协议
喜欢这篇文章?为什么不考虑打赏一下作者呢?
爱发电
文章目录
  1. 题面
    1. 题目描述
    2. 输入格式
    3. 输出格式
    4. 输入输出样例
    5. 数据规模与约定
  2. 思路
  3. 代码