题目
我的解答
用的是暴力解法,思路是新建一个空数组new[],然后逐个比较nums1和nums2各个元素的大小,小的元素加入new中。
LeetCode给的实例有点问题,示例1的nums1就不是有序数组,而且[1,2,3,0,0,0]的长度是6,不是3,我在程序里处理了这个问题,但线上提交就出错。(实在不想修改了)
比较好的答案
思路和我的差不多,只不过我不知道怎么把复杂度控制在O(n),看了答案后才知道我没想到的地方,就是这个while循环的条件:
while p1 < m or p2 < n:
用一句话归纳思路的话,就是:如果遍历nums1的指针到头了,就把nums2剩余的加进去;如果遍历nums2的指针到头了,就把nums1剩余的加进去。
完整的代码:
nums1 = [1,2,3,0,0,0]
nums2 = [2,5,6]
m = 3
n = 3
i = 0
j = 0
new = []
while i < m or j < n:
if i == m:
new.append(nums2[j])
j += 1
elif j == n:
new.append(nums1[i])
i += 1
elif nums1[i] < nums2[j]:
new.append(nums1[i])
i += 1
else:
new.append(nums2[j])
j += 1
print(new)