题目:
给定两个有序整数数组 nums1 和 nums2,将 nums2 合并到 nums1 中,使得 num1 成为一个有序数组。
说明:初始化 nums1 和 nums2 的元素数量分别为 m 和 n。你可以假设 nums1 有足够的空间(空间大小大于或等于 m + n)来保存 nums2 中的元素。
思路:
旧思路就是用一个新的数组,然后两个指针同时遍历两个数组,比较每个位置上然后放入新数组。不过这道题明显不想让你开辟新空间,这个说了nums1有足够空间。一开始的思路是把nums2和nums1比较后插入nums1,插入一位位置后移一位,不过这样就增加了遍历的复杂性。换个角度,从后往前。这样就不怕会覆盖原有的数据。
题解:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
// 检查特殊情况,m=0
if(m==0){
swap(nums1,nums2);
return;//直接返回了
}
int a = m-1, b = n-1, i = nums1.size()-1 ;//初始指针
while((a>=0)&& (b>=0)){ //
//cout<<"mark99999"<<endl;
if (nums1[a] > nums2[b]){
nums1[i] = nums1[a];
cout<<"i:"<<i<<"a:"<<nums1[a]<<endl;
a--;
i--;
}else{
nums1[i] = nums2[b];
cout<<"i:"<<i<<"b:"<<nums2[b]<<endl;
b--;
i--;
}
}
//如果一个数组走到头,但是另一个没有走到头,剩下的都放入
if(b >= 0)
for(;b>=0;b--){
nums1[i] = nums2[b];
i--;
}
// 下面代码多余,因为num1就是本身,不需要把剩下的又放入剩下的位置
//if(a >= 0)
// for(;a>=0;a--){
// nums1[i] = nums1[a];
// i--;
// }
}
更简洁的写法:
if(m == 0)
{
swap(nums1,nums2);
return;
}
int a = m - 1; //指向数组nums1未排序元素末尾
int b = n - 1; //指向数组nums2末尾
int c = m + n - 1; //指向数组nums1末尾
while(a >= 0 && b >= 0)
{
nums1[c--] = (nums1[a] > nums2[b]) ? nums1[a--] : nums2[b--];
}
while(b >=0)
nums1[c--] = nums2[b--];