936. 戳印序列

题目描述:你想要用小写字母组成一个目标字符串 target。 开始的时候,序列由 target.length 个 ‘?’ 记号组成。而你有一个小写字母印章 stamp。在每个回合,你可以将印章放在序列上,并将序列中的每个字母替换为印章上的相应字母。你最多可以进行 10 * target.length 个回合。举个例子,如果初始序列为"???",而你的印章 stamp 是 “abc”,那么在第一回合,你可以得到 “abc??”、"?abc?"、"??abc"。(请注意,印章必须完全包含在序列的边界内才能盖下去。)如果可以印出序列,那么返回一个数组,该数组由每个回合中被印下的最左边字母的索引组成。如果不能印出序列,就返回一个空数组。例如,如果序列是 “ababc”,印章是 “abc”,那么我们就可以返回与操作 “???” -> “abc??” -> “ababc” 相对应的答案 [0, 2];另外,如果可以印出序列,那么需要保证可以在 10 * target.length 个回合内完成。任何超过此数字的答案将不被接受。
解题思路:如果我们按照正序思考的话,前面盖的戳会被后面覆盖,因此我们逆序思考,依次将target变为?,首先盖实际中最后盖上去的一个戳,那就要求我们第一个stamp必须完全和target中的子串匹配,将其变为?之后,问号就可以匹配任意字符。以stamp的长度为窗口从左到右依次滑动,用made和todo分别记录已经匹配上和待匹配的,如果todo为空那么就可以盖上这个戳,改过戳之后的位置都可以匹配上,再更新每个窗口的todo将新增的问号位置删除,用一个队列记录盖过戳之后为问号的位置,循环遍历队列,直到队列为空,代码如下:

class Solution:
    def movesToStamp(self, stamp: str, target: str) -> List[int]:
        target_len = len(target)
        stamp_len = len(stamp)
        done = [False] * target_len
        windows = []
        import queue
        q = queue.Queue()
        ans = []
        for i in range(target_len - stamp_len + 1):
            made, todo = set(), set()
            for j, c in enumerate(stamp):
                if target[i+j] == c:
                    made.add(i+j)
                else:
                    todo.add(i+j)
            windows.append((made, todo))
            if not todo:
                ans.append(i)
                for idx in made:
                    if not done[idx]:
                        q.put(idx)
                        done[idx] = True

        while q.qsize():
            i = q.get()
            for j in range(max(0, i-stamp_len+1), min(i, target_len-stamp_len)+1):
                if i in windows[j][1]:
                    windows[j][1].discard(i)
                    if not windows[j][1] and windows[j][0]:
                        ans.append(j)
                        for m in windows[j][0]:
                            if not done[m]:
                                q.put(m)
                                done[m] = True
        return ans[::-1] if all(done) else []
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值