LeetCode88:合并两个有序数组

给定两个有序整数数组 nums1 和 nums2,将 nums2 合并到 nums1 使得 num1 成为一个有序数组。

说明:

  • 初始化 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]

解析:

       利用从后往前比较的方法,把较大的数据放在后面,依次往前遍历两个数组。

代码:

void merge(vector<int>& nums1, int m, vector<int>& nums2, int n)
{
	int index1 = m - 1;//原nums1的下标
	int index2 = n - 1;
	int index = m + n - 1;//新nums1的下标
	while (index1 >= 0 && index2 >= 0)//从后往前比较,大的先放
	{
		if (nums1[index1] > nums2[index2])
		{
			nums1[index] = nums1[index1];
			--index1;
		}
		else
		{
			nums1[index] = nums2[index2];
			--index2;
		}
		--index;
	}
	while (index2 >= 0)//如果nums1的数据还没遍历完,则顺序刚好
	{
		nums1[index--] = nums2[index2--];
	}
}

leetcode上较为简练的写法

   void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) 
   {
        int p = m-- + n-- - 1;
        while (m >= 0 && n >= 0)
             nums1[p--] = nums1[m] > nums2[n] ? nums1[m--] : nums2[n--];
        while (n >= 0)
            nums1[p--] = nums2[n--]; 
        
    }

猜你喜欢

转载自blog.csdn.net/qq_36214481/article/details/84995536
今日推荐