C++運算符重載(12) - 重載數組索引操作符[]

本篇主要讨論索引操作符 [] 的重載。

下面是關于重載[]的一些注意事項：

1) 當需要檢測下标是否越界時，重載[]是很有用的。

2) 重載函數必須傳回引用，因為類似的表達式“arr[i]”可以被當成一個左值。

下面程式示範了重載索引操作符[]

// 數組類的重載操作
#include<iostream>
#include<cstdlib>
using namespace std;

// 數組類
class Array {
private:
    int* ptr;
    int size;
public:
    Array(int*, int);
    // 重載[]操作符，使得可以像數組一樣通路這個類
    int& operator[] (int);
    void print() const;
};

// []操作符的重載實作。這個函數必須傳回引用，且不能為const，因為可能會用作左值。例如arr[] = xxx;
int& Array::operator[](int index) {
    if (index >= size) {
        cout << "Array index out of bound, exiting";
        exit(0);
    }
    return ptr[index];
}

// 構造函數
Array::Array(int* p = NULL, int s = 0) {
    size = s;
    ptr = NULL;
    if (s != 0) {
        ptr = new int[s];
        for (int i = 0; i < s; i++)
            ptr[i] = p[i];
    }
}

void Array::print() const {
    for (int i = 0; i < size; i++)
        cout << ptr[i] << " ";
    cout << endl;
}

int main() {
    int a[] = { 1, 2, 4, 5 };
    Array arr1(a, 4);
    arr1[2] = 6;
    arr1.print();
    arr1[8] = 6;
    return 0;
}
           

運作結果：

1 2 6 5

Array index out of bound, exiting