《劍指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;
}