面试金典--11.1

题目描述：给定排序后数组A和B，其中A的末端有足够空间存B，编写算法，将B合并到A

思路：

题目的意思应该是不用额外空间，所以从两个数组的最后一个元素往前遍历，将大的放入A的末尾

#include <iostream>
#include <queue>
#include <climits>
#include <algorithm>
#include <memory.h>
#include <stdio.h>
using namespace std;

vector<int> fun(vector<int> a,vector<int> b,int na,int nb)
{
    int lena = a.size();
    int lenb = b.size();

    int i,j;
    int k = lena-1;
    for(i = na-1,j = nb-1 ; i >=0 && j >=0 ;)
    {
        if(a[i] >= b[j])
        {
            a[k] = a[i];
            --i;
            --k;
        }
        else
        {
            a[k] = b[j];
            --j;
            --k;
        }
    }
    if(i < 0)
    {
        while(j >= 0)
        {
            a[k] = b[j];
            --j;
            --k;
        }
    }
    return a;
}

int main()
{
    vector<int> a;
    a.push_back(1);
    a.push_back(2);
    a.push_back(3);
    a.push_back(2);
    a.push_back(2);
    sort(a.begin(),a.end());
    a.push_back(0);

    vector<int> b;
    b.push_back(4);

    vector<int> res = fun(a,b,5,1);
    int i;
    for(i = res.size()-1 ; i >=0 ; --i )
    {
        cout<<res[i]<<endl;
    }
    return 0;
}