leedcode-指针篇(13)

88. 合并两个有序数组

题目

给你两个有序整数数组 nums1 和 nums2,请你将 nums2 合并到 nums1 中,使 nums1 成为一个有序数组。

初始化 nums1 和 nums2 的元素数量分别为 m 和 n 。你可以假设 nums1 的空间大小等于 m + n,这样它就有足够的空间保存来自 nums2 的元素。

示例:
输入:nums1 = [1,2,3,0,0,0], m = 3, nums2 = [2,5,6], n = 3
输出:[1,2,2,3,5,6]

这题的思路,我只能想到用两个指针,然后从头开始比对,num2大于num1的就插入过去,但是后来发现根本不行。只能面向答案去反推思路

class Solution:
	def merge(self,nums1:List[int],m:int,nums2:List[int],n:int)->None:
		s1,s2,end=m-1,n-1,m+n-1
		while(s1>=0 and s2>=0):
			if nums1[s1]<=nums2[s2]:
				nums1[end]=nums2[s2]
				s2-=1
			else: 
				nums1[end]=nums1[s1]
				s1-=1
			end-=1
		if s2>=0:
			nums1[:s2+1]=nums[:s2+1]

思路是:定义三个尾部指针,s1是num1的尾部指针,s2是num2的尾部指针,end是合并后的尾指针。其中s1是指向nums1存在数字的末尾,就比如示例中【1,2,3,0,0,0】,那么s1就是指向3。然后从后往前遍历,如nums2的元素比nums1的元素大,那么就将nums2放到nums1的末尾,也就是
【1,2,3,0,0,6】,然后大数组的指针往前移动;如果nums1的元素大于nums2的元素,那么就将nums1的元素放到nums1的空位置中,如【1,2,3,0,5,6】的时候,nums2=【2,5,6】,而此时s1还是等于3,s2=2,那么,就变成【1,2,3,3,5,6】,然后再比对,发现s1指向的值和s2指向的值相等,那么就变成【1,2,2,3,5,6】。
测试用例:

def merge(nums1,m,nums2,n):
	s1,s2,end=m-1,n-1,m+n-1
	while(s1>=0 and s2>=0):
		if(nums1[s1]<=nums2[s2]):
			nums1[end]=nums2[s2]
			s2-=1
		else: 
			nums1[end]=nums1[s1]
			s1-=1
		end-=1
	if (s2>=0):
		nums1[:s2+1]=nums[:s2+1]
	return nums1

nums=[1,2,3,0,0,0]
result=merge(nums,3,[2,5,6],3)
print(result)
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值