C++ vector用法小记

最近在leetcode上面做题，STL的vector用的甚多，这里稍微的总结下vector的一些常用方法，包括一维的和二维的，暂时能想到的就给记下了。

一维vector

创建一维vector：

vector<int> nums;//不指定长度
vector<int> nums(n); // 指定长度为n 

添加元素

nums.push_back(1);//直接从数组末端添加
nums[i] = 1;//直接赋值给第i个位置

删除元素

nums.resize(nums.size-i); //直接将数组长度减小，某种方式上删掉了后面i个
nums.pop_back();//删掉最后一个元素

数组遍历

for(int i = 0; i < nums.size(); i++){
    cout<<nums[i]<<endl;
}

其他

获得长度：nums.size()

排序(O(nlogn))：sort(nums.begin(),nums.end());

翻转：reverse(nums.begin(), nums.end());

合并两个vector：合并nums1和nums2，并将合并后的数组赋值给nums

vector<int> nums1(m),nums2(n);
vector<int> nums;
nums.resize(m+n);
merge(nums1.begin(), nums1.end(),nums2.begin(),nums2.end(),nums);

二维vector

创建m*n的二维vector: 直接定义

vector<vector <int> > nums(m ,vector<int>(n));    //m*n的二维vector

动态创建m*n的二维vector

方法一：

vector<vector <int> > nums;
nums.resize(m);
for(int i=0;i<m;i++) nums[i].resize(n);

方法二：

vector<vector <int> > nums;
nums.resize(m,vector<int>(n));

动态创建二维数组nums[m][n]

C语言版：

#include<malloc.h>
int **a=(int **)malloc(m*sizeof(int *));
for(int i=0;i<m;i++)
a[i]=(int *)malloc(n*sizeof(int));

C++版：

int **nums=new int*[m];
for(int i=0;i<m;i++) nums[i]=new int[n]; 

初始化二维数组

vector<vector <int> > nums(m ,vector<int>(n,0));    //m*n的二维vector，所有元素为0

获得二维数组的行数：nums.size();

获得二维数组的列数：nums[0].size()

数组遍历

int m = nums.size(),n = nums[0].size();
for(int i = 0; i < m; i++){
    for(int j = 0; j < n; j++){
        cout<<nums[i][j]<<endl;
    }
}