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

题意

这个问题和：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;
}