DFS/BFS
约 1437 个字 135 行代码 预计阅读时间 6 分钟 总阅读量 次
17. 电话号码的数字组合
🔑🔑 难度:Medium 中等
给定一个仅包含数字 2-9 的字符串,返回所有它能表示的字母组合。答案可以按 任意顺序 返回。
给出数字到字母的映射如下(与电话按键相同)。注意 1 不对应任何字母。
class Solution:
def letterCombinations(self, digits: str) -> List[str]:
result = []
n = len(digits)
record = {
"1": "",
"2": "abc",
"3": "def",
"4": "ghi",
"5": "jkl",
"6": "mno",
"7": "pqrs",
"8": "tuv",
"9": "wxyz"
}
def dfs(curr, idx):
if idx == n:
result.append("".join(curr))
return
for char_ in record[digits[idx]]:
dfs(curr + [char_], idx + 1)
if n == 0:
return []
dfs([], 0)
return result
每次枚举一个元素就行了,注意ending condition。
22. 括号生成
🔑🔑 难度:Medium 中等
数字 n 代表生成括号的对数,请你设计一个函数,用于能够生成所有可能的并且 有效的 括号组合。
class Solution:
def generateParenthesis(self, n: int) -> List[str]:
result = []
m = 2 * n
curRoute = [""] * m
def dfs(idx, temp):
if idx == m:
result.append("".join(curRoute))
return
if temp < n:
curRoute[idx] = "("
dfs(idx + 1, temp + 1)
if idx - temp < temp:
curRoute[idx] = ")"
dfs(idx + 1, temp)
dfs(0, 0)
return result
做法:枚举下一个元素可能的括号,同时记录现在已经有的左括号数(temp
)。
什么时候能放左括号?很简单,只要当前已经放的左括号数还没到 n 个就行了;
什么时候能放右括号?这个难理解一点:我只要看目前左边已经放的所有括号 (idx
个)减去目前已经放的左括号数 ( temp
),如果这个数字比 temp
还要小,说明有一些左括号没有闭合,可以放右括号。反过来讲,如果 idx - temp == temp
了,就说明现有的 idx
个括号互相闭合了,不能放右括号。
78. 子集
🔑🔑 难度:Medium 中等
给你一个整数数组 nums
,数组中的元素 互不相同 。返回该数组所有可能的子集(幂集)。
解集 不能 包含重复的子集。你可以按 任意顺序 返回解集。
示例 1:
输入:
nums = [1,2,3]
输出:
[[],[1],[2],[1,2],[3],[1,3],[2,3],[1,2,3]]
class Solution:
def subsets(self, nums: List[int]) -> List[List[int]]:
result = []
n = len(nums)
def backtrack(idx, arr):
result.append(arr)
for j in range(idx, n):
backtrack(j + 1, arr + [nums[j]])
backtrack(0, [])
return result
每次向下走,不带筛除的。
🌟394. 字符串解码
🔑🔑 难度:Medium 中等
示例1:
给定一个经过编码的字符串,返回它解码后的字符串。
编码规则为:
k[encoded_string]
,表示其中方括号内部的encoded_string
正好重复k
次。注意k
保证为正整数。你可以认为输入字符串总是有效的;输入字符串中没有额外的空格,且输入的方括号总是符合格式要求的。
此外,你可以认为原始数据不包含数字,所有的数字只表示重复的次数
k
,例如不会出现像3a
或2[4]
的输入。输入:
s = "3[a]2[bc]"
输出:
"aaabcbc"
class Solution:
def decodeString(self, s: str) -> str:
def dfs(s, idx):
res = ""
multi = 0
while idx < len(s):
if "0" <= s[idx] <= "9" :
multi = multi * 10 + int(s[idx])
elif s[idx] == "[":
idx, tmp = dfs(s, idx + 1)
res += multi * tmp
multi = 0
elif s[idx] == "]":
return idx, res
else:
res += s[idx]
idx += 1
return idx, res
idx, sol = dfs(s, 0)
return sol
第一次写一定会觉得很麻烦不好做的题。第一想法就是递归(虽然题目在stack部分。根据 []
的位置判断递归的情况。遇到 [
就进入下一层,在下一层返回当前的索引位置以及字符串;遇到 ]
就直接停止。
433. 最小基因变化
- 🔑🔑 难度:Medium 中等
基因序列可以表示为一条由 8 个字符组成的字符串,其中每个字符都是 'A'、'C'、'G' 和 'T' 之一。
假设我们需要调查从基因序列 start 变为 end 所发生的基因变化。一次基因变化就意味着这个基因序列中的一个字符发生了变化。
例如,"AACCGGTT" --> "AACCGGTA" 就是一次基因变化。 另有一个基因库 bank 记录了所有有效的基因变化,只有基因库中的基因才是有效的基因序列。(变化后的基因必须位于基因库 bank 中)
给你两个基因序列 start 和 end ,以及一个基因库 bank ,请你找出并返回能够使 start 变化为 end 所需的最少变化次数。如果无法完成此基因变化,返回 -1 。
注意:起始基因序列 start 默认是有效的,但是它并不一定会出现在基因库中。
示例 1:
输入:start = "AACCGGTT", end = "AACCGGTA", bank = ["AACCGGTA"] 输出:1
class Solution:
def minMutation(self, startGene: str, endGene: str, bank: List[str]) -> int:
if startGene == endGene:
return 0
if endGene not in bank:
return -1
queue = deque([startGene])
visited = set([startGene])
step = 0
while queue:
l = len(queue)
for i in range(l):
cur_g = queue.popleft()
if cur_g == endGene:
return step
for j in range(8):
for k in ["A", "G", "C", "T"]:
if cur_g[j] == k:
continue
next_g = cur_g[: j] + k + cur_g[j + 1: ]
if next_g not in visited and next_g in bank:
visited.add(next_g)
queue.append(next_g)
step += 1
return -1
909. 蛇梯棋
- 🔑🔑 难度:Medium 中等
给你一个大小为 n x n 的整数矩阵 board ,方格按从 1 到 n2 编号,编号遵循 转行交替方式 ,从左下角开始 (即,从 board[n - 1][0] 开始)的每一行改变方向。
你一开始位于棋盘上的方格 1。每一回合,玩家需要从当前方格 curr 开始出发,按下述要求前进:
选定目标方格 next ,目标方格的编号在范围 [curr + 1, min(curr + 6, n2)] 。 该选择模拟了掷 六面体骰子 的情景,无论棋盘大小如何,玩家最多只能有 6 个目的地。 传送玩家:如果目标方格 next 处存在蛇或梯子,那么玩家会传送到蛇或梯子的目的地。否则,玩家传送到目标方格 next 。 当玩家到达编号 n2 的方格时,游戏结束。 如果 board[r][c] != -1 ,位于 r 行 c 列的棋盘格中可能存在 “蛇” 或 “梯子”。那个蛇或梯子的目的地将会是 board[r][c]。编号为 1 和 n2 的方格不是任何蛇或梯子的起点。
注意,玩家在每次掷骰的前进过程中最多只能爬过蛇或梯子一次:就算目的地是另一条蛇或梯子的起点,玩家也 不能 继续移动。
举个例子,假设棋盘是 [[-1,4],[-1,3]] ,第一次移动,玩家的目标方格是 2 。那么这个玩家将会顺着梯子到达方格 3 ,但 不能 顺着方格 3 上的梯子前往方格 4 。(简单来说,类似飞行棋,玩家掷出骰子点数后移动对应格数,遇到单向的路径(即梯子或蛇)可以直接跳到路径的终点,但如果多个路径首尾相连,也不能连续跳多个路径) 返回达到编号为 n2 的方格所需的最少掷骰次数,如果不可能,则返回 -1。
class Solution:
def snakesAndLadders(self, board: List[List[int]]) -> int:
n = len(board)
visited = [False] * (n * n + 1)
q = [1]
step = 0
while q:
tmp = q
q = []
for x in tmp:
if x == n * n:
return step
for y in range(x + 1, min(n * n, x + 6) + 1):
r, c = divmod(y - 1, n )
if r % 2:
c = n - 1 - c
nxt = board[-1 - r][c]
if nxt < 0:
nxt = y
if not visited[nxt]:
visited[nxt] = True
q.append(nxt)
step += 1
return -1
重要的是把这个编码规则实现了!!!