leetcode 88 Merge Sorted Array

最新推荐文章于 2024-03-25 21:41:37 发布

原创最新推荐文章于 2024-03-25 21:41:37 发布 · 3.9k 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#algorithm #c++ #leetcode #python

c++ 同时被 3 个专栏收录

79 篇文章

订阅专栏

算法设计与分析

73 篇文章

订阅专栏

leetcode c++ python实现

49 篇文章

订阅专栏

本文介绍了一种用于合并两个已排序数组的算法，并提供了多种实现方式，包括使用多集、迭代器和原地操作等，旨在提高性能并简化代码。

Given two sorted integer arrays nums1 and nums2, merge nums2 intonums1 as one sorted array.

Note:
You may assume that nums1 has enough space (size that is greater or equal tom + n) to hold additional elements from nums2. The number of elements initialized innums1 and nums2 are m and n respectively.

测试用例：

Runtime Error Message:

Last executed input:

Input:[1,2,3,0,0,0], 3, [2,5,6], 3

Output:[1,2,3,5,6]

Expected:[1,2,2,3,5,6]

错误的解决方案：

class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) 
    {
    set<int> result;
    for(int i = 0;i<m;i++)
    {
        result.insert(nums1[i]);
    }
    for(int i = 0;i<n;i++)
    {
        result.insert(nums2[i]);
    }
    nums1.clear();
    set<int>::iterator iter = result.begin();
    for(;iter!=result.end();iter++)
    {
        nums1.push_back(*iter);
    }
    }
};

我的解决方案：上面就是相同的元素没装进来，换成multiset就行了

class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) 
    {
    multiset<int> result;
    for(int i = 0;i<m;i++)
    {
        result.insert(nums1[i]);
    }
    for(int i = 0;i<n;i++)
    {
        result.insert(nums2[i]);
    }
    nums1.clear();
    set<int>::iterator iter = result.begin();
    for(;iter!=result.end();iter++)
    {
        nums1.push_back(*iter);
    }
    }
};

简短的解决方案：

class Solution {
public:
    void merge(int A[], int m, int B[], int n) {
        int k = m + n;
        while (k-- > 0)
            A[k] = (n == 0 || (m > 0 && A[m-1] > B[n-1])) ?  A[--m] : B[--n];
    }
};

可读性较好：

class Solution {
public:
    void merge(int A[], int m, int B[], int n) {
        int i=m-1;
        int j=n-1;
        int k = m+n-1;
        while(i >=0 && j>=0)
        {
            if(A[i] > B[j])
                A[k--] = A[i--];
            else
                A[k--] = B[j--];
        }
        while(j>=0)
            A[k--] = B[j--];
    }
};

python解决方案：

class Solution:
# @param A  a list of integers
# @param m  an integer, length of A
# @param B  a list of integers
# @param n  an integer, length of B
# @return nothing(void)
def merge(self, A, m, B, n):
    x=A[0:m]
    y=B[0:n]
    x.extend(y)
    x.sort()
    A[0:m+n]=x

python解决方案2：thats why we love python

def merge(self, A, m, B, n):
        A[m:] = B[:n]
        A.sort()