模板类的头文件
#pragma once
#include<cassert>
template <typename T>
class Array
{
public:
//构造函数
Array(int sz = 50);
//析构函数
~Array();
//复制构造函数
Array(const Array<T>& a);
//重载 =
Array<T>& operator = (const Array<T>& rhs);
//重载 []
T& operator[] (int i);
const T& operator[] (int i) const;
//重载到T*类型的转换
operator T * ();
operator const T * () const;
//得到数组的大小
int getSize() const;
//改变数组的大小
void resize(int sz);
private:
T* list; //存放动态分配的数组内存的首地址
int size; //数组元素个数
};
//构造函数
template <typename T>
Array<T>::Array(int sz) {
assert(sz >= 0); //判断sz是否大于0
size = sz;
list = new T[size]; //动态分配给size个T类型的空间
}
//析构函数
template <typename T>
Array<T>::~Array() {
delete[] list;
}
//复制构造函数 是在对象刚建立的时候调用,尚未分配内存,所以不需要delete[] list
template <typename T>
Array<T>::Array(const Array<T>& a) {
//深层复制
size = a.size;
list = new T[size];
for (int i = 0;i < size;++i)
list[i] = a.list[i];
}
//重载 = 对象存在了一段时间后调用,对象已经分配内存,需要delete[] list
template <typename T>
Array<T>& Array<T>::operator = (const Array<T>& rhs) {
if (&rhs != this) { //先判断地址是否一样
if (rhs.size != size) { //再判断数组大小是否相同
delete[] list; //如果不同,先删除数组原有内存
size = rhs.size; //设置数组大小
list = new T[size]; //重新分配size个元素的内存
}
for (int i = 0;i < size;++i) //复制
list[i] = rhs.list[i];
}
return *this; //返回当前对象的引用
}
//重载 []
template <typename T>
T& Array<T>::operator[] (int i) {
assert(i >= 0 && i < size);
return list[i];
}
//返回只读引用
template <typename T>
const T& Array<T>::operator[] (int i) const {
assert(i >= 0 && i < size);
return list[i];
}
//重载到T*类型的转换
/*
* 例如void fun(int* p);
* Array<int> a(10);
* fun(a); //这里的a是一个对象,需要有对象到指针类型的转换
*/
template <typename T>
Array<T>::operator T* () {
return list;
}
template <typename T>
Array<T>::operator const T* () const {
return list;
}
//得到数组的大小
template <typename T>
int Array<T>::getSize() const {
return size;
}
//改变数组的大小
template <typename T>
void Array<T>::resize(int sz) {
assert(sz >= 0); //判断sz是否合法
if (sz == size) //如果相等则返回
return;
T* newlist = new T[sz]; //不相等则申请一个新的内存空间
int n = (sz < size) ? sz : size; //令n取sz和size中较小的一个
for (int i = 0;i < n;++i)
newlist[i] = list[i]; //把list[0]到list[n - 1]的元素复制给newlist
delete[] list; //删除list指向的内存空间
list = newlist; //让list指向newlist
size = sz; //把sz赋值给size
}
main.cpp
#include<iostream>
#include<iomanip>
#include"array.h"
using namespace std;
void fun(int* x) {
cout << x[1];
}
int main(void) {
Array<int> a(10);
int n, count = 0;
cin >> n;
for (int i = 2;i < n;i++) {
bool isprime = true;
for (int j = 0;j < count;j++)
if (i % a[j] == 0) {
isprime = false;
break;
}
if (isprime) {
if (count == a.getSize())
a.resize(count * 2);
a[count++] = i;
}
}
for (int i = 0;i < count;i++)
cout << a[i] << ' ';
cout << endl;
return 0;
}
注:模板类的声明文件和实现文件应该放在同一个 .h 文件中,否则会导致程序可以通过编译但无法运行