C++学习笔记——友元及重载运算符

目录

一、友元

1.1声明友元函数

1.2声明友元类

二、运算符重载

2.1重载加号运算符

2.2重载流插入运算符

三、一个简单的银行管理系统

四、 详细的介绍


一、友元

在 C++ 中,友元是一个函数或类,它可以访问另一个类的私有成员或保护成员。通常情况下,我们在类定义中声明友元函数或友元类,以便它们能够访问类的私有成员。友元可以是一个函数、类、或整个命名空间。

1.1声明友元函数

首先,我们来看一下如何声明友元函数。假设我们有一个 Box 类,它表示一个三维立方体:

class Box {
public:
    Box(double l, double w, double h) : length(l), width(w), height(h) {}

private:
    double length;
    double width;
    double height;
};

我们想要计算一个 Box 对象的体积,但是 lengthwidthheight 都是私有成员,无法在类外直接访问。这时候,我们可以使用友元函数来实现。我们可以在类定义中声明一个友元函数,并将其定义为一个全局函数:

class Box {
public:
    Box(double l, double w, double h) : length(l), width(w), height(h) {}

    friend double getVolume(Box box);

private:
    double length;
    double width;
    double height;
};

double getVolume(Box box) {
    return box.length * box.width * box.height;
}

在上面的代码中,我们在 Box 类定义中声明了一个友元函数 getVolume(),并将其定义为一个全局函数。在 getVolume() 函数中,我们可以直接访问 Box 类的私有成员 lengthwidthheight,计算出它的体积并返回。

1.2声明友元类

除了友元函数,我们还可以声明友元类。假设我们有一个 Stack 类,它表示一个栈:

template 
class Stack {
public:
    Stack() : top(-1) {}

private:
    T data[10];
    int top;

    friend class StackIterator;
};

我们想要实现一个迭代器类来遍历栈中的元素。但是,由于 datatop 都是私有成员,无法在迭代器类中直接访问。这时候,我们可以声明 StackIterator 类为 Stack 类的友元类:

template 
class StackIterator {
public:
    StackIterator(Stack& s) : stack(s), index(s.top) {}

    T next() {
        return stack.data[index--];
    }

private:
    Stack& stack;
    int index;
};

template 
class Stack {
public:
    Stack() : top(-1) {}

    friend class StackIterator;

private:
    T data[10];
    int top;
};

在上面的代码中,我们声明了一个友元类 StackIterator,并在 Stack 类定义中将其声明为友元类。在 StackIterator 类中,我们可以直接访问 Stack 类的私有成员 datatop,实现了对栈的迭代。

二、运算符重载

C++ 中有一些运算符是可以被重载的,比如加号、减号、乘号、除号等等。通过重载运算符,我们可以改变它的行为,使其能够作用于我们自定义的类型。

2.1重载加号运算符

假设我们有一个 Complex 类,它表示一个复数:

class Complex {
public:
    Complex(double r = 0.0, double i = 0.0) : real(r), imag(i) {}

private:
    double real;
    double imag;
};

我们想要实现复数的加法运算,但是加号运算符无法直接作用于自定义的类型。这时候,我们可以重载加号运算符 +,使其能够对两个 Complex 对象进行相加,并返回一个新的 Complex 对象:

class Complex {
public:
    Complex(double r = 0.0, double i = 0.0) : real(r), imag(i) {}

    Complex operator+(const Complex& other) const {
        return Complex(real + other.real, imag + other.imag);
    }

private:
    double real;
    double imag;
};

在上面的代码中,我们重载了加号运算符 +,并定义了一个成员函数 operator+()。在函数中,我们定义了一个新的 Complex 对象,其实部为两个复数的实部相加,虚部为两个复数的虚部相加,并将其返回。

2.2重载流插入运算符

除了加号运算符,我们还可以重载其他运算符。比如,我们可以重载流插入运算符 <<,使其能够输出我们自定义的类型。假设我们有一个 Person 类,它表示一个人:

#include 
#include 

class Person {
public:
    Person(std::string n, int a) : name(n), age(a) {}

    friend std::ostream& operator<<(std::ostream& os, const Person& p);

private:
    std::string name;
    int age;
};

std::ostream& operator<<(std::ostream& os, const Person& p) {
    os << "Name: " << p.name << ", Age: " << p.age;
    return os;
}

int main() {
    Person p("Alice", 20);
    std::cout << p << std::endl;
    return 0;
}

在上面的代码中,我们重载了流插入运算符 <<,并将其定义为一个友元函数。在函数中,我们将 Person 对象的姓名和年龄输出到标准输出流 os 中,然后返回该流。在主函数中,我们创建了一个 Person 对象 p,并使用 << 运算符将其输出到标准输出流中。

需要注意的是,在重载运算符时需要遵循一些规则。比如,我们必须使用正确的参数类型和返回值类型,以及保证运算符的语义符合预期。此外,C++ 中还存在一些运算符是无法被重载的,比如条件运算符 ?:,作用域运算符 :: 等等。

三、一个简单的银行管理系统

#include 
#include 
#include 

class Account {
public:
    Account(std::string name, std::string accountNumber, double balance)
        : name(name), accountNumber(accountNumber), balance(balance) {}

    void deposit(double amount) {
        balance += amount;
        std::cout << "Deposit successful. New balance: " << balance << std::endl;
    }

    void withdraw(double amount) {
        if (balance >= amount) {
            balance -= amount;
            std::cout << "Withdrawal successful. New balance: " << balance << std::endl;
        } else {
            std::cout << "Insufficient balance." << std::endl;
        }
    }

    void displayInfo() {
        std::cout << "Name: " << name << std::endl;
        std::cout << "Account Number: " << accountNumber << std::endl;
        std::cout << "Balance: " << balance << std::endl;
    }

private:
    std::string name;
    std::string accountNumber;
    double balance;
};

class Bank {
public:
    void createAccount(std::string name, std::string accountNumber, double balance) {
        Account account(name, accountNumber, balance);
        accounts.push_back(account);
        std::cout << "Account created successfully." << std::endl;
    }

    void deposit(std::string accountNumber, double amount) {
        Account* account = findAccount(accountNumber);
        if (account != nullptr) {
            account->deposit(amount);
        } else {
            std::cout << "Account not found." << std::endl;
        }
    }

    void withdraw(std::string accountNumber, double amount) {
        Account* account = findAccount(accountNumber);
        if (account != nullptr) {
            account->withdraw(amount);
        } else {
            std::cout << "Account not found." << std::endl;
        }
    }

    void displayAllAccounts() {
        for (const auto& account : accounts) {
            account.displayInfo();
            std::cout << "---------------------------" << std::endl;
        }
    }

private:
    std::vector accounts;

    Account* findAccount(std::string accountNumber) {
        for (auto& account : accounts) {
            if (account.getAccountNumber() == accountNumber) {
                return &account;
            }
        }
        return nullptr;
    }
};

int main() {
    Bank bank;
    int choice;

    do {
        std::cout << "1. Create Account" << std::endl;
        std::cout << "2. Deposit" << std::endl;
        std::cout << "3. Withdraw" << std::endl;
        std::cout << "4. Display All Accounts" << std::endl;
        std::cout << "5. Exit" << std::endl;
        std::cout << "Enter your choice: ";
        std::cin >> choice;

        if (choice == 1) {
            std::string name, accountNumber;
            double balance;
            std::cout << "Enter name: ";
            std::cin >> name;
            std::cout << "Enter account number: ";
            std::cin >> accountNumber;
            std::cout << "Enter initial balance: ";
            std::cin >> balance;
            bank.createAccount(name, accountNumber, balance);
        } else if (choice == 2) {
            std::string accountNumber;
            double amount;
            std::cout << "Enter account number: ";
            std::cin >> accountNumber;
            std::cout << "Enter amount to deposit: ";
            std::cin >> amount;
            bank.deposit(accountNumber, amount);
        } else if (choice == 3) {
            std::string accountNumber;
            double amount;
            std::cout << "Enter account number: ";
            std::cin >> accountNumber;
            std::cout << "Enter amount to withdraw: ";
            std::cin >> amount;
            bank.withdraw(accountNumber, amount);
        } else if (choice == 4) {
            bank.displayAllAccounts();
        }

    } while (choice != 5);

    return 0;
}

在上面的代码中,我们定义了两个类:AccountBankAccount 类代表一个银行账户,具有姓名、账号和余额属性,以及存款、取款和展示信息的方法。Bank 类代表整个银行系统,具有创建账户、存款、取款和显示所有账户的方法。其中,存款和取款的操作会调用 Account 类相应的方法。

在主函数中,我们通过简单的菜单来模拟用户与银行系统的交互。用户可以选择创建账户、存款、取款或显示所有账户的操作,直到选择退出。

四、 详细的介绍

以下是代码中每个部分的更详细解释:

class Account {
public:
    Account(std::string name, std::string accountNumber, double balance)
        : name(name), accountNumber(accountNumber), balance(balance) {}

Account 类的构造函数中,我们使用传入的参数来初始化私有成员变量 nameaccountNumberbalance

    void deposit(double amount) {
        balance += amount;
        std::cout << "Deposit successful. New balance: " << balance << std::endl;
    }

deposit 方法用于存款操作。它接受一个 double 类型的参数 amount,代表要存入的金额。方法内部,我们将传入的金额加到余额上,并输出新的余额。

    void withdraw(double amount) {
        if (balance >= amount) {
            balance -= amount;
            std::cout << "Withdrawal successful. New balance: " << balance << std::endl;
        } else {
            std::cout << "Insufficient balance." << std::endl;
        }
    }

withdraw 方法用于取款操作。它接受一个 double 类型的参数 amount,代表要取出的金额。方法内部,我们首先检查账户余额是否足够,如果足够就从余额中减去取款金额,并输出新的余额。如果余额不足,则输出错误信息。

    void displayInfo() {
        std::cout << "Name: " << name << std::endl;
        std::cout << "Account Number: " << accountNumber << std::endl;
        std::cout << "Balance: " << balance << std::endl;
    }

displayInfo 方法用于展示账户的信息。它输出账户的姓名、账号和余额。

class Bank {
public:
    void createAccount(std::string name, std::string accountNumber, double balance) {
        Account account(name, accountNumber, balance);
        accounts.push_back(account);
        std::cout << "Account created successfully." << std::endl;
    }

createAccount 方法用于创建一个新的账户。它接受三个参数:姓名、账号和初始余额。方法内部,我们创建一个新的 Account 对象,并将其添加到 accounts 列表中。然后,我们输出成功创建账户的消息。

    void deposit(std::string accountNumber, double amount) {
        Account* account = findAccount(accountNumber);
        if (account != nullptr) {
            account->deposit(amount);
        } else {
            std::cout << "Account not found." << std::endl;
        }
    }

deposit 方法用于给指定账户存款。它接受两个参数:账户号码和存款金额。方法内部,我们通过调用 findAccount 方法来查找要操作的账户。如果找到了账户,则调用该账户的 deposit 方法执行存款操作。如果未找到账户,则输出错误信息。

    void withdraw(std::string accountNumber, double amount) {
        Account* account = findAccount(accountNumber);
        if (account != nullptr) {
            account->withdraw(amount);
        } else {
            std::cout << "Account not found." << std::endl;
        }
    }

withdraw 方法用于给指定账户取款。它接受两个参数:账户号码和取款金额。方法内部,我们通过调用 findAccount 方法来查找要操作的账户。如果找到了账户,则调用该账户的 withdraw 方法执行取款操作。如果未找到账户,则输出错误信息。

    void displayAllAccounts() {
        for (const auto& account : accounts) {
            account.displayInfo();
            std::cout << "---------------------------" << std::endl;
        }
    }

displayAllAccounts 方法用于展示银行中所有账户的信息。它遍历 accounts 列表,并对每个账户调用 displayInfo 方法来输出信息。在每个账户信息输出之后,我们打印分隔线以区分不同的账户。

int main() {
    Bank bank;
    int choice;

    do {
        std::cout << "1. Create Account" << std::endl;
        std::cout << "2. Deposit" << std::endl;
        std::cout << "3. Withdraw" << std::endl;
        std::cout << "4. Display All Accounts" << std::endl;
        std::cout << "5. Exit" << std::endl;
        std::cout << "Enter your choice: ";
        std::cin >> choice;

        if (choice == 1) {
            std::string name, accountNumber;
            double balance;
            std::cout << "Enter name: ";
            std::cin >> name;
            std::cout << "Enter account number: ";
            std::cin >> accountNumber;
            std::cout << "Enter initial balance: ";
            std::cin >> balance;
            bank.createAccount(name, accountNumber, balance);
        } else if (choice == 2) {
            std::string accountNumber;
            double amount;
            std::cout << "Enter account number: ";
            std::cin >> accountNumber;
            std::cout << "Enter amount to deposit: ";
            std::cin >> amount;
            bank.deposit(accountNumber, amount);
        } else if (choice == 3) {
            std::string accountNumber;
            double amount;
            std::cout << "Enter account number: ";
            std::cin >> accountNumber;
            std::cout << "Enter amount to withdraw: ";
            std::cin >> amount;
            bank.withdraw(accountNumber, amount);
        } else if (choice == 4) {
            bank.displayAllAccounts();
        }

    } while (choice != 5);

    return 0;
}

main 函数中,我们使用一个循环来模拟用户与银行系统的交互。我们显示一个简单的菜单,让用户选择不同的操作。根据用户的选择,我们调用 Bank 类的相应方法来执行操作。当用户选择退出时,循环结束,程序终止。

C++学习笔记——友元及重载运算符_第1张图片

你可能感兴趣的:(学习,笔记)