STL_员工分组案例

 //main.cpp文件
 #include "worker.h"
 #include "manager.h"
 ​
 int main()
 {
     manager m;
     m.show();
     return 0;
 }
 //work.h文件
 #pragma once
 #include <string>
 #include <map>
 #include <vector>
 #include <map>
 using namespace std;
 ​
 class worker
 {
 public:
     string getName();
     void setName(string name);
 ​
     string getTelephone();
     void setTelephone(string telephone);
 ​
     int getAge();
     void setAge(int age);
 ​
     float getSalary();
     void setSalary(float salary);
 ​
 private:
     string name;
     int age;
     string telephone;
     float salary;
 };
 //work.cpp文件
 #include "worker.h"
 #include <string>
 using namespace std;
 ​
 string worker::getName() {
     return this->name;
 }
 ​
 void worker::setName(string name) {
     this->name = name;
 }
 ​
 string worker::getTelephone() {
     return this->telephone;
 }
 ​
 void worker::setTelephone(string telephone) {
     this->telephone = telephone;
 }
 ​
 int worker::getAge() {
     return this->age;
 }
 ​
 void worker::setAge(int age) {
     this->age = age;
 }
 ​
 float worker::getSalary() {
     return this->salary;
 }
 ​
 void worker::setSalary(float salary) {
     this->salary = salary;
 }
 //manager.h文件
 #pragma once
 ​
 #define WORKER_NUMBER 5
 ​
 #define SALE_DEPATMENT 1 //销售部门
 #define DEVELOP_DEPATMENT 2 //研发部门
 #define FINACIAL_DEPATMENT 3 //财务部门
 ​
 #include <string>
 #include <map>
 #include <vector>
 #include <map>
 #include "worker.h"
 using namespace std;
 class manager
 {
 public:
     void create_worker();
     void divide_worker();
     void print_worker(int departID);
     void print_worker_by_group();
     void show();
 private:
     vector<worker> worker_vector;
     multimap<int, worker> worker_group;
 };
 //manager.cpp文件
 #include "worker.h"
 #include "manager.h"
 #include <string>
 #include <map>
 #include <vector>
 #include <iostream>
 #include <map>
 #include <time.h>
 using namespace std;
 ​
 void manager::create_worker() {
 ​
     string name_seed = "ABCDE";
 ​
     for (int i = 0; i < WORKER_NUMBER; i++) {
 ​
         worker worker;
         string tmp = name_seed.substr(i, 1);
         worker.setName("选手" + tmp);
         worker.setAge(rand() % 15 + 20);
         worker.setTelephone("010-88888888");
         worker.setSalary(rand()%10000+1000.00);
 ​
         this->worker_vector.push_back(worker);
     }
 }
 ​
 void manager::divide_worker() {
     srand(time(NULL));
     for (vector<worker>::iterator it = this->worker_vector.begin() ; it !=this->worker_vector.end(); it++) {
 ​
         int departID = rand() % 3 + 1;
         switch (departID) {
 ​
             case SALE_DEPATMENT:
                 this->worker_group.insert(make_pair(SALE_DEPATMENT, *it));
                 break;
             case DEVELOP_DEPATMENT:
                 this->worker_group.insert(make_pair(DEVELOP_DEPATMENT, *it));
                 break;
             case FINACIAL_DEPATMENT:
                 this->worker_group.insert(make_pair(FINACIAL_DEPATMENT, *it));
                 break;
             default:
                 break;
         }
     }
 }
 ​
 void manager::print_worker(int departID) {
     
     multimap<int, worker>::iterator it = this->worker_group.find(departID);
     
     int DepartCount = this->worker_group.count(departID);
     int num = 0;
     for (multimap<int, worker>::iterator pos = it; it != this->worker_group.end() && num < DepartCount; pos++, num++) {
         cout << "姓名:" << pos->second.getName()<< " 年龄:" << pos->second.getAge() << " 电话:" << pos->second.getTelephone() << " 工资:" << pos->second.getSalary() << endl;
     }
 }
 ​
 //打印每一部分员工信息
 void manager::print_worker_by_group() {
 ​
     //显示销售部门
     cout << "销售部门:" << endl;
     this->print_worker(SALE_DEPATMENT);
     //显示开发部门
     cout << "研发部门:" << endl;
     this->print_worker(DEVELOP_DEPATMENT);
     //显示财务部门
     cout << "财务部门:" << endl;
     this->print_worker(FINACIAL_DEPATMENT);
 }
 ​
 void manager::show() {
     this->create_worker();
     this->divide_worker();
     this->print_worker_by_group();
 }