前言📢📢
Python每日一练来啦,本文已收录于:《Python每日一练》专栏
此专栏目的在于,帮忙学习Python的小白提高编程能力,训练逻辑思维,持续更新中,欢迎免费订阅!!!
*
1. 问题描述
给你两个按 非递减顺序 排列的整数数组 nums1 和 nums2,另有两个整数 m 和 n ,分别表示 nums1 和 nums2 中的元素数目。
请你 合并 nums2 到 nums1 中,使合并后的数组同样按 非递减顺序 排列。
注意: 最终,合并后数组不应由函数返回,而是存储在数组 nums1 中。为了应对这种情况,nums1 的初始长度为 m + n,其中前 m 个元素表示应合并的元素,后 n 个元素为 0 ,应忽略。nums2 的长度为 n 。
示例 1:
输入: nums1 = [1,2,3,0,0,0], m = 3, nums2 = [2,5,6], n = 3输出: [1,2,2,3,5,6]解释: 需要合并 [1,2,3] 和 [2,5,6] 。合并结果是 [1 ,2 ,2,3,5,6] ,其中斜体加粗标注的为 nums1 中的元素。
示例 2:
输入: nums1 = [1], m = 1, nums2 = [], n = 0输出: [1]**解释:**需要合并 [1] 和 [] 。合并结果是 [1] 。
示例 3:
输入: nums1 = [0], m = 0, nums2 = [1], n = 1输出: [1]**解释:**需要合并的数组是 [] 和 [1] 。合并结果是 [1] 。注意,因为 m = 0 ,所以 nums1 中没有元素。nums1 中仅存的 0 仅仅是为了确保合并结果可以顺利存放到 nums1 中。
提示:
-
nums1.length == m + n
-
nums2.length == n
-
0 <= m, n <= 200
-
1 <= m + n <= 200
-
-109 <= nums1[i], nums2[j] <= 109
2. 问题分析
因为两个数组nums1和nums2都是有序的,并且nums1后n个元素是0,最终返回的结果是nums1,所以考虑将数据从nums1、nums2的最后开始比较,逐一放入nums1数组中,如果最后nums2还剩余有数据,直接拷贝到nums1中。
*
3. 算法思路
思路:
(1)定义三个指针
nums1_index: 指向nums1数组有效数据的末尾(m-1)
nums2_index: 指向nums2数组有序数据的末尾(n-1)
nums1_final_index: 指向nums1数组的末尾(m+n-1)
(2)从后向前,数据进行比较nums1[nums1_index], nums2[nums2_index]; 将大的数据放到nums1[nums1_final_index],并移动响应的指针
(3)如果最后nums2还剩有数据,直接拷贝到nums1中。
*
4. 代码实现
python
from typing import List
class Solution:
def merge(self, nums1: List[int], m: int, nums2: List[int], n: int) -> None:
nums1_index = m - 1
nums2_index = n - 1
nums1_final_index = m + n - 1
while nums1_index >= 0 and nums2_index >= 0:
if nums1[nums1_index] > nums2[nums2_index]:
nums1[nums1_final_index] = nums1[nums1_index]
nums1_index -= 1
else:
nums1[nums1_final_index] = nums2[nums2_index]
nums2_index -= 1
nums1_final_index -= 1
while nums2_index >= 0:
nums1[nums1_final_index] = nums2[nums2_index]
nums2_index -= 1
nums1_final_index -= 1
if __name__ == '__main__':
Solution().merge([1], 1, [0], 0)