93. 复原 IP 地址
照猫画虎,自己捣鼓AC了,但是没有加入任何剪枝:
class Solution:
def restoreIpAddresses(self, s: str) -> List[str]:
res = []
parts = []
def backtrack(start):
if start >= len(s):
if len(parts) == 4:
res.append('.'.join(parts))
return
for end in range(start + 1, len(s) + 1):
subString = s[start:end]
if int(subString) <= 255 and str(int(subString)) == subString:
parts.append(subString)
backtrack(end)
parts.pop()
backtrack(0)
return res
剪枝:
- 如果已经有4部分或剩余字符不足以完成4部分,可以提前终止。
- 如果当前部分超过3位数字,可以提前跳出循环,因为IP地址的每个部分最多3位数字。
- 如果剩余字符太少或太多,无法形成有效的IP地址,可以跳过当前迭代。
class Solution:
def restoreIpAddresses(self, s: str) -> List[str]:
res = []
parts = []
def backtrack(start):
# 如果已经有4部分且到达字符串末尾,提前终止
if len(parts) == 4:
if start == len(s):
res.append('.'.join(parts))
return
# 如果剩余字符太少或太多,无法形成有效的IP地址
if len(s) - start > 3 * (4 - len(parts)) or len(s) - start < (4 - len(parts)):
return
for end in range(start + 1, min(start + 4, len(s) + 1)):
subString = s[start:end]
if int(subString) <= 255 and (subString == "0" or not subString.startswith("0")):
parts.append(subString)
backtrack(end)
parts.pop()
backtrack(0)
return res
78. 子集
回溯,在每次递归中,首先将path
的副本添加到结果 res
中。
class Solution:
def subsets(self, nums: List[int]) -> List[List[int]]:
res = []
path = []
def backtrack(start):
res.append(path[:])
if start >= len(nums):
return
for i in range(start, len(nums)):
path.append(nums[i])
backtrack(i + 1)
path.pop()
backtrack(0)
return res
好简单,真不容易..
90. 子集 II
又会!(发疯
class Solution:
def subsetsWithDup(self, nums: List[int]) -> List[List[int]]:
res = []
path = []
def backtrack(start):
res.append(path[:])
for i in range(start, len(nums)):
if i > start and nums[i] == nums[i - 1]:
continue
path.append(nums[i])
backtrack(i + 1)
path.pop()
nums.sort()
backtrack(0)
return res
今日总结:
三题都会,信心增强。
历经毒打之后感受到了一丝温暖。