手撕vector
Myclass.h
#pragma once
#include<iostream>
#include<Windows.h>
#define SUCCESS 1 // 成功
#define ERROR -1 // 失败
#define MALLOC_ERROR -2 // 申请内存失败
#define INDEX_ERROR -3 // 错误的索引号
template <class T_ELE>
class Vector
{
public:
Vector();
Vector(DWORD dwSize);
~Vector();
public:
DWORD at(DWORD dwIndex, OUT T_ELE* pEle); //根据给定的索引得到元素
DWORD push_back(T_ELE Element); //将元素存储到容器最后一个位置
BOOL pop_back(); //删除最后一个元素
DWORD insert(DWORD dwIndex, T_ELE Element); //向指定位置新增一个元素
DWORD capacity(); //返回在不增容的情况下，还能存储多少元素
VOID clear(); //清空所有元素
BOOL empty(); //判断Vector是否为空 返回true时为空
DWORD erase(DWORD dwIndex); //删除指定元素
DWORD size(); //返回Vector元素数量的大小
private:
BOOL expand();
private:
DWORD m_dwIndex; //下一个可用索引
DWORD m_dwIncrement; //每次增容的大小
DWORD m_dwLen; //当前容器的长度
DWORD m_dwInitSize; //默认初始化大小
T_ELE* m_pVector; //容器指针
};
//构造函数（不带参数）
template <class T_ELE>
Vector<T_ELE>::Vector()
:m_dwInitSize(10), m_dwIncrement(5)
{
//1.创建长度为m_dwInitSize个T_ELE对象
m_pVector = new T_ELE[m_dwInitSize];
//2.将新创建的空间初始化
memset(m_pVector, 0, m_dwInitSize * sizeof(T_ELE));
//3.设置其他值
m_dwLen = m_dwInitSize;
m_dwIndex = 0;
}
//构造函数（带参数）
template <class T_ELE>
Vector<T_ELE>::Vector(DWORD dwSize)
:m_dwIncrement(5)
{
//1.创建长度为dwSize个T_ELE对象
m_pVector = new T_ELE[dwSize];
//2.将新创建的空间初始化
memset(m_pVector, 0, dwSize * sizeof(T_ELE));
//3.设置其他值
m_dwLen = dwSize;
m_dwIndex = 0;
}
//析构函数
template <class T_ELE>
Vector<T_ELE>::~Vector()
{
//释放空间 delete[]
delete[] m_pVector;
m_pVector = NULL;
}
template <class T_ELE>
BOOL Vector<T_ELE>::expand()
{
DWORD dwTempLen = 0;
T_ELE* pTemp = NULL;
// 1. 计算增加后的长度
dwTempLen = m_dwLen + m_dwIncrement;
// 2. 申请空间
pTemp = new T_ELE[dwTempLen];
// 3. 将数据复制到新的空间
memcpy(pTemp,m_pVector,sizeof(T_ELE)*m_dwLen);
// 4. 释放原来空间
delete[] m_pVector;
m_pVector = pTemp;
pTemp = NULL;
// 5. 为各种属性赋值
m_dwLen = dwTempLen;
return TRUE;
}
//将元素存储到容器最后一个位置
template <class T_ELE>
DWORD Vector<T_ELE>::push_back(T_ELE Element)
{
//1.判断是否需要增容，如果需要就调用增容的函数
if (m_dwIndex >= m_dwLen)
{
expand();
}
//2.将新的元素复制到容器的最后一个位置
memcpy(&m_pVector[m_dwIndex],&Element,sizeof(T_ELE));
//3.修改属性值
m_dwIndex++;
return SUCCESS;
}
//删除最后一个元素
template<class T_ELE>
BOOL Vector<T_ELE>::pop_back() {
/*if (m_dwLen > 0) {
memset(m_pVector + m_dwLen, 0, sizeof(T_ele));
m_dwLen--;
}
*/
//下面这个是删除当前元素中最后一个元素的方法，上面注释那个是删除整个容器最后一个容器
if (m_dwIndex > 0) {
memset(m_pVector + (m_dwIndex - 1), 0, sizeof(T_ELE));
m_dwIndex--;
return TRUE;
}
}
//向指定位置新增一个元素
template <class T_ELE>
DWORD Vector<T_ELE>::insert(DWORD dwIndex, T_ELE Element)
{
//1.判断索引是否在合理区间
if(dwIndex<0|| dwIndex>m_dwIndex)
{
return INDEX_ERROR;
}
//2.判断是否需要增容，如果需要就调用增容的函数
if(dwIndex>=m_dwLen)
{
expand();
}
//3.将dwIndex只后的元素后移
for(int i = m_dwIndex;i> dwIndex;i--)
{
memcpy(&m_pVector[i], &m_dwIndex[i - 1], sizeof(T_ELE));
}
//4.将Element元素复制到dwIndex位置
memcpy(&m_dwIndex[dwIndex],&Element,sizeof(T_ELE));
//5.修改属性值
m_dwIndex++;
}
//根据给定的索引得到元素
template <class T_ELE>
DWORD Vector<T_ELE>::at(DWORD dwIndex, T_ELE* pEle)
{
//判断索引是否在合理区间
if (dwIndex<0 || dwIndex>=m_dwIndex)
{
return INDEX_ERROR;
}
//将dwIndex的值复制到pEle指定的内存
memcpy(&pEle, &m_pVector[dwIndex], sizeof(T_ELE));
}
//删除指定元素
template <class T_ELE>
DWORD Vector<T_ELE>::erase(DWORD dwIndex)
{
//1.判断索引是否在合理区间
if (dwIndex <0 || dwIndex >= m_dwIndex || dwIndex > m_dwLen)
{
return INDEX_ERROR;
}
//2.将dwindex之后的元素前移
for (int i = dwIndex; i < m_dwIndex - 1; i++)
{
memcpy(&m_pVector[i], &m_pVector[i + 1], sizeof(T_ELE));
}
//3.修改属性值
m_dwIndex--;
}
//判断Vector是否为空 返回true时为空
template<class T_ELE>
BOOL Vector<T_ELE>::empty() {
if (m_pVector == NULL or m_dwIndex = 0) {
return true;
}
return false;
}
template<class T_ELE>
DWORD Vector<T_ELE>::size() {
return m_dwLen;
}
//查询剩余容量
template<class T_ELE>
DWORD Vector<T_ELE>::capacity() {
if (m_pVector)
{
return m_dwLen - m_dwIndex;
}
return ERROR;
}
template<class T_ELE>
void Vector<T_ELE>::clear() {
memset(m_pVector, 0, sizeof(T_ELE) * m_dwLen);
m_dwIndex = 0;
}
Vector.cpp
#include"MyClass.h"
#include"malloc.h"
void TestVector()
{
Vector<int>* pVector = new Vector<int>(5);
pVector->push_back(1);
pVector->push_back(2);
pVector->push_back(3);
pVector->push_back(4);
pVector->push_back(5);
//pVector->push_back(6);
pVector->erase(3);
}
int main(int argc, char* argv[])
{
TestVector();
return 0;
}
补充一下用vector来定义二维数组
初始化一个二维vector数组并指定大小的方法有:
1. vector<vector<int>> vec(行数, vector<int>(列数));
// 如 vector<vector<int>> vec(3,vector<int>(4)) 定义的就是3行4列（有没有想到汤家凤hhh）的二维数组
2. vector<vector<int>> vec(行数, vector<int>(列数, 初始值));
//如 vector<vector<int>> vec(3,vector<int>(4，1)) 定义的就是3行4列（有没有想到汤家凤hhh）的二维数组，并且二维数组里的数全部初始化为1
3. vector<vector<int>> vec; vec.resize(行数); vec[i].resize(列数); vec[i] = {值};
//这就说明可以定义不同行的列数不相同，可以指定某行的列数
