C++STL中vector使用策略(二)

下面来看一道LeetCode上的困难题——两个排序数组的中位数

       这道题只要掌握了vector的基本用法以及常用的成员函数,做起来不难,就是要细心

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

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。

输出格式

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

样例输入

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<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;	 
}

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

扫码关注云+社区