operator[] w wersji z referencją

program.cpp

@@ -8,10 +8,12 @@ int main() {
     w.push_back(10);
     w.push_back(11);
     w.push_back(12);
-    std::cout << "Element nr 0: " << w.get(0) << '\n';
-    std::cout << "Element nr 2: " << w.get(2) << '\n';
-    w.set(2, 22);
-    std::cout << "Element nr 2: " << w.get(2) << '\n';
+    std::cout << "Element nr 0: " << w[0] << '\n';
+    std::cout << "Element nr 2: " << w[2] << '\n';
+    // w.set(2, 2222);
+    w[2] = 2220;
+    w[2] += 2;
+    std::cout << "Element nr 2: " << w[2] << '\n';
     std::cout <<"size: " << w.size() << '\n';
     std::cout << '\n';
 
@@ -27,7 +29,7 @@ int main() {
     // EXN w.set(50, 5050);
 
     try {
-        w.set(50, 5050);
+        w[50] = 5050;
         std::cout << "udało się ustawić element 50\n";
     } catch(const char *exn) {
         std::cout << "wyjątek " << exn << '\n';

wektor.cpp

@@ -32,21 +32,14 @@ void Wektor::push_back(TypWartosci e) {
     t[liczba_elementow++] = e;
 }
 
-void Wektor::set(TypIndeksu idx, TypWartosci e) {
-    if(idx < 0 || idx >= liczba_elementow) {
-        throw "indeks poza zakresem";
-    }
-    t[idx] = e;
-}
-
-TypWartosci Wektor::get(TypIndeksu idx) {
+TypWartosci& Wektor::operator[](TypIndeksu idx) {
     if(idx < 0 || idx >= liczba_elementow) {
         throw "indeks poza zakresem";
     }
     return t[idx];
 }
 
-std::ostream& operator<<(std::ostream& out, const Wektor& w) {
+std::ostream&  operator<<(std::ostream& out, const Wektor& w) {
     out << '[';
     for(TypIndeksu i=0; i<w.liczba_elementow; i++) {
         out << w.t[i] << ", ";

wektor.h

@@ -27,11 +27,10 @@ public:
     /** Dodanie nowego elementu na końcu. */
     void push_back(TypWartosci e);
 
-    /** Przypisanie nowej wartości na podanej pozycji. */
-    void set(TypIndeksu idx, TypWartosci e);
-
     /** Odczyt wartości spod podanej pozycji. */
-    TypWartosci get(TypIndeksu idx);
+
+    /** Dostęp do wartości na podanej pozycji. */
+    TypWartosci& operator[](TypIndeksu idx);
 
     friend std::ostream& operator<<(std::ostream&, const Wektor&);
 };