题解：洛谷 P1381 单词背诵

【题目来源】

洛谷：P1381 单词背诵 - 洛谷

【题目描述】

灵梦有 \(n\) 个单词想要背，但她想通过一篇文章中的一段来记住这些单词。

文章由 \(m\) 个单词构成，她想在文章中找出连续的一段，其中包含最多的她想要背的单词（重复的只算一个）。并且在背诵的单词量尽量多的情况下，还要使选出的文章段落尽量短，这样她就可以用尽量短的时间学习尽可能多的单词了。

【输入】

第 \(1\) 行一个数 \(n\)，接下来 \(n\) 行每行是一个长度不超过 \(10\) 的字符串，表示一个要背的单词。

接着是一个数 \(m\)，然后是 \(m\) 行长度不超过 \(10\) 的字符串，每个表示文章中的一个单词。

【输出】

输出文件共 \(2\) 行。第 \(1\) 行为文章中最多包含的要背的单词数，第 \(2\) 行表示在文章中包含最多要背单词的最短的连续段的长度。

【输入样例】

3
hot
dog
milk
5
hot
dog
dog
milk
hot

【输出样例】

3
3

【算法标签】

《洛谷 P1381 单词背诵》 #模拟# #字符串# #二分# #哈希,hash# #双指针,two-pointer# #福建省历届夏令营#

【代码详解】

#include <bits/stdc++.h>
using namespace std;

const int MAX_N = 100005;  // 定义最大单词数量
int n, m;                  // n: 目标单词数量, m: 文章单词数量
int sum = 0, len = 0;      // sum: 包含的目标单词种类数, len: 最短包含所有目标单词的段落长度
string s[MAX_N];           // 存储文章中的单词序列
map<string, bool> word;    // 记录目标单词集合
map<string, int> cnt;      // 记录当前窗口中各目标单词出现次数

int main()
{
    // 输入目标单词
    cin >> n;
    for (int i = 1; i <= n; i++)
    {
        string s1;
        cin >> s1;
        word[s1] = true;   // 标记为目标单词
    }

    // 输入文章单词并处理
    cin >> m;
    for (int i = 1, j = 1; j <= m; j++)  // i: 窗口左边界, j: 窗口右边界
    {
        cin >> s[j];  // 输入当前单词

        // 如果是目标单词，更新计数
        if (word[s[j]])
        {
            cnt[s[j]]++;
            // 如果是第一次出现该目标单词，更新种类数
            if (cnt[s[j]] == 1)
            {
                sum++;
                len = j - i + 1;  // 初始化段落长度
            }
        }

        // 调整窗口左边界
        while (i <= j)
        {
            // 如果当前左边界单词只出现一次，不能移动
            if (cnt[s[i]] == 1)
                break;

            // 如果左边界单词出现多次，可以右移
            if (cnt[s[i]] >= 2)
            {
                cnt[s[i]]--;
                i++;
            }

            // 如果左边界单词不是目标单词，直接右移
            if (!word[s[i]])
                i++;
        }

        // 更新最短段落长度
        len = min(len, j - i + 1);
    }

    // 输出结果
    cout << sum << endl << len;

    return 0;
}

【运行结果】

3
hot
dog
milk
5
hot
dog
dog
milk
hot
3
3