刷题-关键字标识

题目描述

给定一个关键词集合 words 和一个字符串 inputStr，要求将 inputStr 中出现的所有关键词用标签标记：

• 关键词标签的起始为<b>,结束为</b>。
• 把可以合并的标签进行合并，即使用最少的标签。合并规则如下：
  • 关键词相邻则进行合并，如 ab、cd为关键词，且在字符串中相邻，则合并为 <b>abcd</b>
  • 关键词相交则进行合并，如 zhi、hid为关键词，且在字符串中相交，则合并为 <b>zhid</b>

解答要求时间限制：1000ms, 内存限制：256MB

输入

第1行一个整数 count，表示 words 中的关键词的个数，取值范围：[1, 64]
第2行 count 个字符串，表示关键字词列表 words，每个关键词仅含英文小写字母，长度范围：[1,16]
第3行一个字符串 inputStr，仅含英文小写字母（无空格），长度范围：[1,512]

输出

一个用最少关键词标签标记的字符串

样例

输入样例 1 

4
cd df op qr
opqracdfg

输出样例 1

<b>opqr</b>a<b>cdf</b>g

提示样例 1

• 关键词 op qr 出现在字符串中，且在字符串中相邻，因此可合并为 opqr；
• cd df 出现在字符串中，且在字符串中相交，因此可合并为 cdf ；
• 其它字符保持不变。

处理后为 <b>opqr</b>a<b>cdf</b>g，使用了 2 个标签 。
<b>op</b><b>qr</b>a<b>cd</b><b>df</b>g也是一种标记方式，但关键词未作合并，需要用到 4 个标签，标签个数不是最少的。

 

输入样例 2 

3
abbb def bbg
aabbbgz

输出样例 2

a<b>abbbg</b>z

提示样例 2

关键字abbb和bbg合并为abbbg

解题代码

#include <iostream>
#include <sstream>
#include <string>
#include <vector>
#include <utility>
#include <algorithm>
using namespace std;

class Solution {
public:
    // 待实现函数，在此函数中填入答题代码;
    static bool cmp(const pair<int, int> &a, const pair<int, int> &b)
    {
        if (a.first == b.first)
            return a.second < b.second;
        return a.first < b.first;
    }
    
    string GetTaggedString(const vector<string> &words, const string &inputStr)
    {
        string str = inputStr;
        vector<pair<int, int>> keyPos;

        //找到所有关键字在str的位置, 记录在keyPos中
        for (string s : words) {
            int pos = 0;
            int index = 0;
            while ((index = str.find(s, pos)) != string::npos) {
                keyPos.push_back({index, index+s.size()});
                pos = index + s.size(); 
            }
        }
        
        if (keyPos.size() == 0)
            return str;

        //按位置升序排列keyPos，用于后续合并
        sort(keyPos.begin(), keyPos.end(), cmp);

        int left = keyPos[0].first;
        int right = keyPos[0].second;
        vector<pair<int, int>> myPos;
        
        //keyPos区间合并, 跳过重叠字符位置
        for (int i = 1; i < keyPos.size(); i++) {
            pair<int, int> p = keyPos[i];
            if (p.first == right) {
                right = p.second;
            } else if (p.first < right) {
                right = max(right, p.second);
            } else {
                myPos.push_back({left, right});
                left = p.first;
                right = p.second;
            }
        }
        myPos.push_back({left, right});


        //根据myPos中记录的关键词位置，插入新字符
        string start = "<b>";
        string end = "</b>";

        for (int i = 0; i < myPos.size(); i++) {
            left = myPos[i].first + i*(start.size() + end.size());
            right = myPos[i].second + i*(start.size() + end.size()) + start.size();
            str.insert(left, start);
            str.insert(right, end);
        }

        return str;
    }
};

inline int ReadInt()
{
    int number;
    std::cin >> number;
    return number;
}

void SplitString(const string &input, char sperChar, vector<string> &outArray)
{
    stringstream sstr(input);
    string token;
    while (getline(sstr, token, sperChar)) {
        outArray.push_back(token);
    }
}

inline string ReadLine()
{
    string line;
    getline(cin, line);
    return line;
}

int main()
{   
    int wordsNum = ReadInt();
    cin.ignore();
    vector<string> words;
    string wordsList = ReadLine();
    SplitString(wordsList, ' ', words);
    words.resize(wordsNum);
    string inputStr = ReadLine();
    
    Solution solu;
    string result = solu.GetTaggedString(words, inputStr);
    cout << result;
    return 0;
}