LeetCode 616. 给字符串添加加粗标签（Trie树）

文章目录

- 1. 题目
- 2. 解题

1. 题目

给一个字符串 s 和一个字符串列表 dict ，你需要将在字符串列表中出现过的 s 的子串添加加粗闭合标签 <b> 和 </b> 。
如果两个子串有重叠部分，你需要把它们一起用一个闭合标签包围起来。
同理，如果两个子字符串连续被加粗，那么你也需要把它们合起来用一个加粗标签包围。

样例 1：
输入：
s = "abcxyz123"
dict = ["abc","123"]
输出：
"<b>abc</b>xyz<b>123</b>"样例 2：
输入：
s = "aaabbcc"
dict = ["aaa","aab","bc"]
输出：
"<b>aaabbc</b>c"注意：
给定的 dict 中不会有重复的字符串，且字符串数目不会超过 100 。
输入中的所有字符串长度都在范围 [1, 1000] 内。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/add-bold-tag-in-string
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

跟 LeetCode 758. 字符串中的加粗单词（Trie树）一模一样，改下变量名，字符集改成128大小即可

class trie
{
public:trie* next[128] = {NULL};bool isend =  false;int count = 0;void insert(string& s){trie* cur = this;for(int i = 0; i < s.size(); i++){if(cur->next[s[i]] == NULL)cur->next[s[i]] = new trie();cur = cur->next[s[i]];}cur->count++;cur->isend = true;}
};
class Solution {
public:string addBoldTag(string S, vector<string>& words) {trie *t = new trie(), *cur;for(auto& w : words)t->insert(w);vector<bool> bold(S.size(), false);//加黑标记int boldl = 0, boldr=-1;//开始加粗的位置l,rfor(int i = 0, j; i < S.size(); ++i){cur = t;boldl = max(boldl, i);//加黑的地方左端点j = i;while(j < S.size() && cur && cur->next[S[j]]){cur = cur->next[S[j]];if(cur->isend)boldr = j;//可以加黑的右端点j++;}while(boldl <= boldr)bold[boldl++] = true;//标记加黑}string ans;for(int i = 0; i < S.size(); ++i){if((i==0 && bold[i]) || (i>0 && !bold[i-1] && bold[i]))//i起点ans += "<b>";ans += S[i];if((i==S.size()-1 && bold[i]) || (i<S.size()-1 && bold[i] && !bold[i+1]))//i是终点ans += "</b>";}return ans;}
};