面试题 10.01 合并两个有序数组(双指针)

题目描述
给你两个有序整数数组 nums1 和 nums2,请你将 nums2 合并到 nums1 中,使 nums1 成为一个有序数组。
初始化 nums1 和 nums2 的元素数量分别为 m 和 n 。你可以假设 nums1 的空间大小等于 m + n,这样它就有足够的空间保存来自 nums2 的元素。

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

题目已知nums1 可以存储下两个数组中所有的元素,因此无需使用多余的数组,题目中给定的是两个有序数组,那么可以利用这个性质来遍历数组,起始指针的位置处于实际元素的最末尾位置,示例1 中 nums1 数组指针开始 i 指向3位置,nums2 指针 j 开始指向6位置,此外还需要一个指针来标记结果数组中遍历到的位置,起始位置为 m+n-1 处。

  • 如果说指针 j 所指向的元素大于 i 指针指向的元素,就需要将大的元素放置到指针 k 处,j 向左移动,k 也向左移动,同理对于i 指针也是如此
  • 需要注意的是如果两个数组遍历过程中其中一个数组提前遍历结束,那么需要将另一个数组的元素全部拷贝到剩下的位置中去。也就是 i 有可能等于-1,j 有可能等于-1
  • 为什么不需要判断k>=0 呢?原因是两个数组遍历结束,k 恰好等于0,k-- 之后等于-1,但是循环已经结束了,所以不需要判断该条件。
class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
        int k = m+n-1;
        if(!m) nums1=nums2;
        int i=m-1,j=n-1;
        while(i>=0 || j>=0)
        {
            if(i==-1)  nums1[k--]=nums2[j--]; 
            else if(j==-1)  nums1[k--]=nums1[i--];
            else if(nums2[j]>nums1[i]) nums1[k--]=nums2[j--];    
            else nums1[k--]=nums1[i--];
           
        }
    }
};

上述代码合并一下可以得到下面的代码。

class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
       int count=n+m-1;
       n=n-1,m=m-1;
       while(n>=0 || m>=0)
       {
           if(m>=0 && A[m]>B[n] || n<0) A[count--]=A[m--];
           
           else A[count--]=B[n--];
       }
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值