陣列類别範本

模板類的標頭檔案

#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 檔案中,否則會導致程式可以通過編譯但無法執行