数据结构程序设计实验三

//[问题描述] 每个员工的信息包括：编号、姓名、性别、出生年月、学历、职务、电话、住址等。
//系统能够完成员工信息的查询、更新、插入、删除、排序等功能。
//[基本要求] （1） ok排序：按不同关键字，对所有员工的信息进行排序。（2） ok查询：按特定条件查找员工。 
//（3） ok更新：按编号对某个员工的某项信息进行修改。 （4） ok插入：加入新员工的信息。 （5） ok删除：按编号删除已离职的员工的信息。



#include<iostream>
#include<cstdio>
#include<string>
#include<vector>
#include<algorithm>
using namespace std;

class Management {
public:
    int ID;//编号
    string name;//姓名
    string sex;//性别
    string birth;//出生年月
    string diploma;//学历
    string job;//职务
    string tele;//电话
    string address;//住址
public:
    Management(){}
    Management(int id,string Name,string Sex,string Birth,string Diploma,
        string Job,string Tele,string Address):ID(id),name(Name),sex(Sex),
        birth(Birth),diploma(Diploma),job(Job),tele(Tele),address(Address){}

    friend ostream& operator<<(ostream& out, const Management& e);
    friend istream& operator>>(istream& in, Management& e);
};

istream& operator>>(istream& in, Management& e)
{
    cout << "输入ID："; cin >> e.ID;cout << "输入姓名："; cin >> e.name;
    cout << "输入性别："; cin >> e.sex; cout << "输入出生年月："; cin >> e.birth;
    cout << "输入学历："; cin >> e.diploma; cout << "输入职务："; cin >> e.job; 
    cout << "输入电话："; cin >> e.tele; cout << "输入住址："; cin >> e.address;
    return in;//输入编号、姓名、性别、出生年月、学历、职务、电话、住址
}

ostream& operator<<(ostream& out, const Management& e)
{
    cout << "ID："; out<< e.ID<<endl; cout << "姓名："; out << e.name << endl;
    cout << "性别："; out << e.sex << endl; cout << "出生年月："; out << e.birth << endl;
    cout << "学历："; out << e.diploma << endl; cout << "职务："; out << e.job << endl;
    cout << "电话："; out << e.tele << endl; cout << "住址："; out << e.address << endl;
    return out;//输出
}

class EmployeeManagement :public Management {
private:
    vector<Management>employees;
public:
    void addEmployee(const Management& e) {//添加员工
        employees.push_back(e);
    }

    void deleteEmployee(int id) {//删除员工
        auto it = remove_if(employees.begin(), employees.end(), [id](const Management& e) {
            return e.ID == id;
            });//将对应ID员工放在容器末尾
        employees.erase(it, employees.end());
    }

    void updateEmployee(int id) {//更新员工信息
        for (auto& e : employees){
            if (e.ID == id){
                Management ne;
                cout << "输入该员工新信息：\n";
                cin >> ne;
                e = ne;
            }
        }
        cout << "未找到员工信息！\n";
    }

    void findEmployee(int id) {//按id查找员工
        for (const auto& e : employees){
            if (e.ID == id){
                cout << e;
                return;
            }
        }
        cout << "未找到员工信息！\n";
    }

    void sortEmployees(int option) {//按不同方式排序
        if (option == 1) {
            // 按编号排序
            sort(employees.begin(), employees.end(), [](const Management& e1, const Management& e2) {
                return e1.ID < e2.ID;
                });
        }
        else if (option == 2) {
            // 按姓名排序
            sort(employees.begin(), employees.end(), [](const Management& e1, const Management& e2) {
                return e1.name < e2.name;
                });
        }
        else if (option == 3) {
            // 按职位排序
            sort(employees.begin(), employees.end(), [](const Management& e1, const Management& e2) {
                return e1.job < e2.job;
                });
        }
        cout << "排序完成！\n";
    }

    void displayEmoployees()const{//显示所有员工信息
        if (employees.empty()) {
            cout << "没有员工信息！\n";
            return;
        }
        int i = 1;
        for (const auto& e : employees){
            cout << string(20, '-') << "第" << i << "位员工信息" << string(20, '-') << endl;
            cout << e ;
            i++;
        }
    }
};

void menu()
{
    cout << "\n员工管理系统\n";
    cout << "1.添加员工\n";
    cout << "2.删除员工\n";
    cout << "3.更新员工信息\n";
    cout << "4.按ID查找员工\n";
    cout << "5.对员工排序\n";
    cout << "6.显示所有员工\n";
    cout << "0.退出系统\n";
    cout << "请输入选择：";
}

int main()
{
    EmployeeManagement manager;
    int choice;
    while (true)
    {
        menu();
        cin >> choice;
        if (choice == 0)
            break;
        switch (choice) {
        case 1: {
            Management e;
            cout << "输入要添加的员工信息：\n";
            cin >> e;
            manager.addEmployee(e);
            break;
            }
        case 2: {
            int id;
            cout << "输入要删除的员工ID：\n";
            cin >>id;
            manager.deleteEmployee(id);
            break;
        }
        case 3: {
            int id;
            cout << "输入要更新的员工ID：\n";
            cin >> id;
            manager.updateEmployee(id);
            break;
        }
        case 4:{
            int id;
            cout << "输入要查找的员工ID：\n";
            cin >> id;
            manager.findEmployee(id);
            break;
        }
        case 5: {
            int option;
            cout << "1.按编号排序\n";
            cout << "2.按姓名排序\n";
            cout << "3.按职位排序\n";
            cout << "输入选择的排序方式：";
            cin >> option;
            manager.sortEmployees(option);
            break;
        }
        case 6: {
            cout << "所有员工信息如下：\n";
            manager.displayEmoployees();
            break;
        }

        }
    }
    return 0;
}