Add Bold Tag in String — LeetCode 616 Python Solution

MediumLeetCode PremiumTrieArrayHash TableStringString Matching
Problem
#616
Pattern
Trie
Reading time
10 min

The problem

You are given a string s and an array of strings words. You should add a closed pair of bold tag <b> and </b> to wrap the substrings in s that exist in words.

Example

Input
s = "abcxyz123", words = ["abc","123"]
Output
"<b>abc</b>xyz<b>123</b>"
Explanation
The two strings of words are substrings of s as following: "abcxyz123".

Python solution

Python
class Trie:
    def __init__(self):
        self.children = [None] * 128
        self.is_end = False

    def insert(self, word):
        node = self
        for c in word:
            idx = ord(c)
            if node.children[idx] is None:
                node.children[idx] = Trie()
            node = node.children[idx]
        node.is_end = True


class Solution:
    def addBoldTag(self, s: str, words: List[str]) -> str:
        trie = Trie()
        for w in words:
            trie.insert(w)
        n = len(s)
        pairs = []
        for i in range(n):
            node = trie
            for j in range(i, n):
                idx = ord(s[j])
                if node.children[idx] is None:
                    break
                node = node.children[idx]
                if node.is_end:
                    pairs.append([i, j])
        if not pairs:
            return s
        st, ed = pairs[0]
        t = []
        for a, b in pairs[1:]:
            if ed + 1 < a:
                t.append([st, ed])
                st, ed = a, b
            else:
                ed = max(ed, b)
        t.append([st, ed])

        ans = []
        i = j = 0
        while i < n:
            if j == len(t):
                ans.append(s[i:])
                break
            st, ed = t[j]
            if i < st:
                ans.append(s[i:st])
            ans.append('<b>')
            ans.append(s[st : ed + 1])
            ans.append('</b>')
            j += 1
            i = ed + 1

        return ''.join(ans)

Complexity

MeasureComplexity
TimeO(n)
SpaceO(n) auxiliary

Pattern: Trie

Store a set of words by their shared prefixes so lookups cost the length of the word. LeetCode 616. Add Bold Tag in String is filed here because LeetCode tags it Trie, which is the vocabulary this hub collects.

The trie guide has the Python template for the pattern and the 49 LeetCode problems that use it.

Related problems

Frequently asked questions

How hard is LeetCode 616. Add Bold Tag in String?
LeetCode 616. Add Bold Tag in String is rated Medium on LeetCode.
What is the time complexity of LeetCode 616. Add Bold Tag in String?
The Python solution on this page runs in O(n).
What is the space complexity of LeetCode 616. Add Bold Tag in String?
The Python solution on this page uses O(n) auxiliary space.
What topics does LeetCode 616. Add Bold Tag in String cover?
LeetCode 616. Add Bold Tag in String is tagged Trie, Array, Hash Table, String and String Matching on LeetCode.
Is LeetCode 616. Add Bold Tag in String a premium problem?
Yes. LeetCode 616. Add Bold Tag in String is a LeetCode Premium problem, so the full statement and test cases require a paid LeetCode subscription.

Stuck on problems like this in a live interview?

Stealth Interview is a desktop app for macOS and Windows. It reads the problem off your screen and returns a working solution with a step-by-step explanation and its time and space complexity — invisible to screen sharing.

Get Stealth Interview