前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >LeetCode 758. 字符串中的加粗单词(Trie树)

LeetCode 758. 字符串中的加粗单词(Trie树)

作者头像
Michael阿明
发布2020-07-13 16:16:10
1.1K0
发布2020-07-13 16:16:10
举报
文章被收录于专栏:Michael阿明学习之路

1. 题目

给定一个关键词集合 words 和一个字符串 S,将所有 S 中出现的关键词加粗。所有在标签 <b> 和 </b> 中的字母都会加粗。

返回的字符串需要使用尽可能少的标签,当然标签应形成有效的组合。

例如,给定 words = ["ab", "bc"] 和 S = "aabcd",需要返回 "a<b>abc</b>d"。注意返回 "a<b>a<b>b</b>c</b>d" 会使用更多的标签,因此是错误的。

代码语言:javascript
复制
注:
words 长度的范围为 [0, 50]。
words[i] 长度的范围为 [1, 10]。
S 长度的范围为 [0, 500]。
所有 words[i] 和 S 中的字符都为小写字母。

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

2. 解题

  • 将集合里的单词全部插入trie树
  • 以S的每个位置为起点在trie树开始查找完整单词,记录可以加黑的地方,标记在bool数组里
代码语言:javascript
复制
class trie
{
public:
    trie* next[26] = {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]-'a'] == NULL)
                cur->next[s[i]-'a'] = new trie();
            cur = cur->next[s[i]-'a'];
        }
        cur->count++;
        cur->isend = true;
    }
};
class Solution {
public:
    string boldWords(vector<string>& words, string S) {
    	trie *t = new trie(), *cur;
    	for(auto& w : words)
    		t->insert(w);
    	vector<bool> bold(S.size(), false);//加黑标记
    	int boldl = 0, boldr=-1;//开始加粗的位置l,r
    	for(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]-'a'])
    		{
    			cur = cur->next[S[j]-'a'];
    			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;
    }
};

12 ms 11.2 MB

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2020/07/03 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 1. 题目
  • 2. 解题
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档