《剑指offer》合并两个有序的数组

来源:互联网 发布:python参考手册 编辑:程序博客网 时间:2024/06/03 19:46

题意

这个问题和:http://blog.csdn.net/jacketinsysu/article/details/52303266,是差不多的,都是在给定的空间上做操作。

给定两个升序的数组,v1有足够的空间可以容易v1和v2的所有元素。要求将v1和v2的元素合并成一个升序的数组,并且时间复杂度为O(n+m),而空间复杂度为O(1)。

思路

从大的开始比较和”入队“。
比如1 3 5和2 4合并的话,从尾部比较:
5比4大,5放到最后边;
3比4小,4放到5的前面一格;
3比2大,3放到4的前面一格;
1比2小,2放到3的前面一格;
1,没人来比较,1放到2的前面一格。
END

代码

void mergeSortedArray(vector<int>& v1, const vector<int>& v2, int size1) {    int allIndex = size1 + v2.size() - 1, index1 = size1 - 1, index2 = v2.size() - 1;    // 经典的写法是while (index >= 0 && index2 >= 0)    // 不过我觉得这样子写更简洁!    while (index1 >= 0 || index2 >= 0) {        if (index1 < 0)            v1[allIndex--] = v2[index2--];        else if (index2 < 0)            v1[allIndex--] = v1[index1--];        else {            if (v1[index1] >= v2[index2])                v1[allIndex--] = v1[index1--];            else                v1[allIndex--] = v2[index2--];        }    }}

完整的示例代码:

#include <iostream>#include <vector>#include <algorithm>using namespace std;void mergeSortedArray(vector<int>& v1, const vector<int>& v2, int size1) {    int allIndex = size1 + v2.size() - 1, index1 = size1 - 1, index2 = v2.size() - 1;    while (index1 >= 0 || index2 >= 0) {        if (index1 < 0)            v1[allIndex--] = v2[index2--];        else if (index2 < 0)            v1[allIndex--] = v1[index1--];        else {            if (v1[index1] >= v2[index2])                v1[allIndex--] = v1[index1--];            else                v1[allIndex--] = v2[index2--];        }    }}vector<int> genData(size_t size) {    static const int MAX = 100;    vector<int> res(size);    for (size_t i = 0; i < size; ++i) {        res[i] = rand() % MAX;    }    sort(res.begin(), res.end());    return res;}void display(const vector<int>& nums) {    for (int i = 0; i < nums.size(); ++i)        cout << nums[i] << ' ';    cout << endl;}int main() {    srand(time(0));    vector<int> v1 = genData(5);    vector<int> v2 = genData(6);    display(v1);    display(v2);    v1.insert(v1.end(), v2.begin(), v2.end());    mergeSortedArray(v1, v2, 5);    display(v1);    return 0;}
0 0