题目描述:你想要用小写字母组成一个目标字符串 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 []