LintCode-Merge Two Sorted Arrays

最新推荐文章于 2022-04-28 23:08:10 发布

Immortal_s

最新推荐文章于 2022-04-28 23:08:10 发布

阅读量260

点赞数

CC 4.0 BY-SA版权

分类专栏： LintCode

本文链接：https://blog.youkuaiyun.com/qq_24889575/article/details/81542613

LintCode 专栏收录该内容

20 篇文章

订阅专栏

本文介绍了一个简单的算法，用于将两个已排序的整数数组合并成一个新的有序数组。通过使用双指针技巧，该方法能够有效地处理数组大小不一的情况。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Description

Merge two given sorted integer array A and B into a new sorted integer array.

Example

A=[1,2,3,4]

B=[2,4,5,6]

return [1,2,2,3,4,4,5,6]

Challenge

How can you optimize your algorithm if one array is very large and the other is very small?

题目模板

class Solution {
public:
  /**
   * @param A: sorted integer array A
   * @param B: sorted integer array B
   * @return: A new sorted integer array
   */
  vector<int> mergeSortedArray(vector<int> &A, vector<int> &B) {
      // write your code here
  }
};

题目大意

给你两个已经排好序的不定长数组vector，让你把这两个数组的数按从小到大的顺序合并到一个不定长数组vector里。

大概思路

题目很简单，和数据结构指针那一章的课后习题一样。需要注意的就是当一个数组读到尾的时候，另一个数组可能还有数据，最后需要判断下。

class Solution {
public:
  /**
   * @param A: sorted integer array A
   * @param B: sorted integer array B
   * @return: A new sorted integer array
   */
  vector<int> mergeSortedArray(vector<int> &A, vector<int> &B) {
      // write your code here
      vector<int> after;
      int a = 0, b = 0;
      int lena = A.size(), lenb = B.size();
      while(a != lena && b != lenb){
          if(A[a] < B[b])
              after.push_back(A[a++]);
          else
              after.push_back(B[b++]);
      }
      if(a == lena)
          for(; b<lenb; b++)
              after.push_back(B[b]);
      if(b == lenb)
          for(; a<lena; a++)
              after.push_back(A[a]);

      return after;
  }
};