C++STL中vector使用策略（二）

下面来看一道LeetCode上的困难题

题目链接：两个排序数组的中位数

题解

首先利用 push_back 函数，将 nums2 依次插入到 nums1 尾部，然后利用sort(nums1.begin(),nums2.end())函数将合并后的元素进行排序。最后利用 size() 判断其元素个数是偶数还是奇数，以此决定输出中间两位数的平均数还是中间的数。

代码

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2)
    {
        for(int its = 0;its < nums2.size();++its)
        {
            nums1.push_back(nums2.at(its));
        }
        sort(nums1.begin(),nums1.end());
        if(nums1.size() % 2 == 0)
        {
            return (double)(nums1.at((nums1.size()/2) - 1) + nums1.at(nums1.size()/2))/2;
        }
        else
        {
            return nums1.at(((nums1.size())/2));
        }
    }
};

打印锯齿矩阵

锯齿矩阵是指每一行包含的元素个数不相同的矩阵，比如：
3 5 2 6 1
2 3 4
1 6 2 7

读入若干对整数（x,y），表示在第 x 行的末尾加上一个元素 y。输出最终的锯齿数组。初始时矩阵为空。

输入格式

第一行输入两个整数 (1≤n,m≤10000)，其中 n 表示锯齿数组的行数，m 表示插入的元素总数。接下来一共 mm 行，每行两个整数 x,y(1≤x≤n,0≤y≤10000)，表示在第 x 行的末尾插入一个元素 y。

输出格式

一共输出 行，每行若干个用空格分隔的整数。如果某行没有任何元素，则输出一个空行。

样例输入

3 12
1 3
2 2
2 3
2 4
3 1
3 6
1 5
1 2
1 6
3 2
3 7
1 1

样例输出

3 5 2 6 1
2 3 4
1 6 2 7

题解

这题刚开始一看可能感觉需要用二维向量vector<vector <int> >vec，但是其实一个vector<int> vec[1000]就能满足这题的要求了，但是要先分清楚这几个的区别

vector<vector<int> >vec;//vector里面每个元素都是vector，相当于二维vector
vector<int>vec[1000];//开一个1000的数组，里面每个元素是一个vector
vector<int>vec(1000);//开一个vector，大小1000

代码

#include <bits/stdc++.h>
using namespace std;
int main()
{
    vector<int> vec[10005];
    int i,j,n,m,tmp,ans = 0,x,y;
    cin>>n>>m;
    for(i = 0;i < m;i++)
    {
        cin>>x>>y;
        vec[x].push_back(y);
    }        
    for(int i = 1;i <= n;i++)
    {
        for(j = 0;j < vec[i].size();j++)
        {
            if(j != vec[i].size() - 1)
            {
                cout<<vec[i][j]<<" "; 
            }
            else
            {
                cout<<vec[i][j];
            }
        }
        if(i < n)
        {
            cout<<endl;
        }
    }
    return 0;     
}