3.10 - 案例-员工分组

一、案例描述

• 公司招聘了10个员工（ABCDEFGHIJ),10名员工进入公司后，需要指派员工到各个部门工作
• 员工信息有：姓名、薪水； 部门分为：策划、美术、研发
• 随机给10名员工分配部门和薪水
• 通过multimap进行信息的插入 key（部门编号） value（员工）
• 分部门显示员工信息

二、实现步骤

1. 创建10名员工，放到vector中
2. 遍历vector容器，取出每个员工，进行随机分组
3. 分组后，将员工部门编号作为key，具体员工作为value，放入到multimap容器中
4. 分部门显示员工信息

#include <iostream>
#include <map>
#include<vector>
#include<string>
#include<ctime>
using namespace std;

#define CEHUA 0
#define MEISHU 1
#define YANFA 2


class Worker
{
public:
    string m_name;
    int m_salary;
};

//1、创建员工
void createWorker(vector<Worker>& v)
{
    string nameSeed = "ABCDEFGHIJ";
    for (int i = 0; i < 10; i++)
    {
        Worker work;
        work.m_name = "员工";
        work.m_name += nameSeed[i];

        work.m_salary = rand() % 10000 + 10000;   // 10000 ~19999

        //将员工放入到容器中
        v.push_back(work);
    }
}

//2、员工分组
void  setGroup(vector<Worker>& v, multimap<int , Worker>& m)
{
    for (vector<Worker>::iterator it = v.begin(); it != v.end(); it++)
    {
        //产生随机部门编号
        int deptid = rand() % 3; // 0 1 2

        //将员工插入的分组中
        //key部门编号，value具体员工
        m.insert(make_pair(deptid, *it));
    }
}

//3、分组显示员工
void showWorkerByGroup(multimap<int , Worker>&m )
{
    //0 : A B C    1:D  E   2:F G
    cout << "策划部门： " << endl;
    multimap<int ,Worker>::iterator pos = m.find(CEHUA);
    int count = m.count(CEHUA);     //统计具体人数
    int index = 0;
    for (; pos != m.end() && index < count; pos++, index++)
    {
        cout << "姓名： " << (*pos).second.m_name << " 薪水：" << (*pos).second.m_salary << endl;
    }

    cout << "---------------------------------" << endl;
    cout << "美术部门： " << endl;
    pos = m.find(MEISHU);
    count = m.count(MEISHU);     //统计具体人数
    index = 0;
    for (; pos != m.end() && index < count; pos++, index++)
    {
        cout << "姓名： " << (*pos).second.m_name << " 薪水：" << (*pos).second.m_salary << endl;
    }

    cout << "---------------------------------" << endl;
    cout << "研发部门： " << endl;
    pos = m.find(YANFA);
    count = m.count(YANFA);     //统计具体人数
    index = 0;
    for (; pos != m.end() && index < count; pos++, index++)
    {
        cout << "姓名： " << (*pos).second.m_name << " 薪水：" << (*pos).second.m_salary << endl;
    }
}


int main()
{
    srand((unsigned int)time(NULL));
    //1、创建员工
    vector<Worker> v;
    createWorker(v);

    //2、员工分组
    multimap<int, Worker> m;
    setGroup(v, m);

    //3、分组显示员工
    showWorkerByGroup(m);


    测试
    //for (vector<Worker>::iterator it = v.begin(); it != v.end(); it++)
    //{
    //    cout << "姓名： " << (*it).m_name << " 工资： " << (*it).m_salary << endl;
    //}

}