23. 合并K个升序链表
给你一个链表数组,每个链表都已经按升序排列。
请你将所有链表合并到一个升序链表中,返回合并后的链表。
示例 1:
输入:lists = [[1,4,5],[1,3,4],[2,6]]
输出:[1,1,2,3,4,4,5,6]
解释:链表数组如下:
[
1->4->5,
1->3->4,
2->6
]
将它们合并到一个有序链表中得到。
1->1->2->3->4->4->5->6
【思路】
合并两个链表的升级版,最简单的思路是两两合并,遍历列表。高效做法:分而治之。对这个列表进行分治合并链表操作。
【实现】
class Solution:
def mergeKLists(self, lists: List[ListNode]) -> ListNode:
# 合并两个链表
def merge2(l1, l2):
res = ListNode(None)
node = res
while l1 and l2:
if l1.val<l2.val:
node.next,l1 = l1,l1.next
else:
node.next,l2 = l2,l2.next
node = node.next
if l1:
node.next = l1
else:
node.next = l2
return res.next
n = len(lists)
# 分治
def merge(left, right):
if left > right:
return
if left == right:
return lists[left]
mid = (left + right) // 2
l1 = merge(left, mid)
l2 = merge(mid + 1, right)
return merge2(l1, l2)
return merge(0,n-1)
26. 删除排序数组中的重复项
给定一个排序数组,你需要在 原地 删除重复出现的元素,使得每个元素只出现一次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
【思路】
该题目需要在原始的列表上修改,假设不重复项有K个,数组为N,则该数组前K项为不重复项,后面N-K项维持不动。最简单的方法是利用切片完成,进阶方法可以使用快慢指针,快指针遍历整个列表,慢指针记录与快指针相同时的值。
【实现】
class Solution:
def removeDuplicates(self, nums: List[int]) -> int:
if len(nums) < 2:
return 1
i, j = 0, 1
while j < len(nums):
if nums[i] != nums[j]:
i += 1
nums[i] = nums[j]
else:
j += 1
return i + 1
33. 搜索旋转排序数组
升序排列的整数数组 nums 在预先未知的某个点上进行了旋转(例如, [0,1,2,4,5,6,7] 经旋转后可能变为 [4,5,6,7,0,1,2] )。
请你在数组中搜索 target ,如果数组中存在这个目标值,则返回它的索引,否则返回 -1 。
示例 1:
输入:nums = [4,5,6,7,0,1,2], target = 0
输出:4
【思路】
看到有序–>立马推二分查找。由于这道题只是部分有序,故需要增加是否有序的判断。注意边界条件的判别。
【实现】
class Solution:
def search(self, nums: List[int], target: int) -> int:
if len(nums) == 0:
return -1
l = 0
r = len(nums) - 1
while l < r:
mid = l + (r - l) // 2
if nums[mid] < nums[r]: # [mid, r]有序
if nums[mid] < target <= nums[r]:
l = mid + 1
else:
r = mid
else: # [l, mid]有序
if nums[l] <= target <= nums[mid]:
r = mid
else:
l = mid + 1
return -1 if nums[l] != target else l