題目描述
【leetcode】88. 合併兩個有序數組( Merge Sorted Array )
給定兩個有序整數數組 nums1 和 nums2,將 nums2 合併到 nums1 中,使得 num1 成爲一個有序數組。
說明:
初始化 nums1 和 nums2 的元素數量分別爲 m 和 n。
你可以假設 nums1 有足夠的空間(空間大小大於或等於 m + n)來保存 nums2 中的元素。
第一次解答
思路:
nums1已經有足夠的空間,
數組從小到大排序
爲了防止大量的內存拷貝,從後往前添加元素
test case:
[1,2,3,0,0,0]
3
[2,5,6]
3
[4,5,6,0,0,0]
3
[1,2,3]
3
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
// 序號指向兩個數組末尾
int index_1 = m-1;
int index_2 = n-1;
for(int i=m+n-1; i>=0; --i){
// 如果一個數組序號小於0,那隻添加另一個數組元素
if(0 > index_1){
nums1[i] = nums2[index_2--];
continue;
}
if(0 > index_2){
nums1[i] = nums1[index_1--];
continue;
}
// 若序號都大於0,挑選數值大的元素添加
if(nums1[index_1] > nums2[index_2]){
nums1[i] = nums1[index_1--];
}
else{
nums1[i] = nums2[index_2--];
}
}
}
};
結果:
第二次解答
思路:
方法與第一次解答一樣,不變,優化代碼減少if判斷次數。
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
// 序號指向兩個數組末尾
int index_1 = m-1;
int index_2 = n-1;
int index = m+n-1;
while(0 <= index_1 && 0 <= index_2){
// 若序號都大於0,挑選數值大的元素添加
nums1[index--] = (nums1[index_1] > nums2[index_2] ? nums1[index_1--] : nums2[index_2--]);
}
// 到這裏後,若0 > index_2,則merge完成
// 若0 > index_1,則把nums2剩下元素放到nums1中即可
while(0 <= index_2){
nums1[index--] = nums2[index_2--];
}
}
};
結果: