STL案例2-员工分组

案例：十个员工（ABCDEFGHIJ），需要指定在哪个部门工作。

员工信息有：姓名、工资组成。部分分为：策划、美术、研发，随机给10名员工分配部门和工资。

通过multimap进行信息的插入key（部门编号）value（员工）

分部门显示员工信息

步骤：

1、创建10名员工，放到vector中

2、遍历vector容器，取出每个员工，进行随机分组（录入信息、随机数）

3、分组后，将员工部门编号作为key，具体员工作为value，放到multimap容器中

4、分部门显示员工信息

代码实现：

/*
 * STL案例2：员工分组
 */

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

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

class Worker
{
public:
    string m_Name;
    int m_Salary;
};

void createWorker(vector<Worker>& v)
{
    string nameSeed = "ABCDEFGHIJ";
    for (int i = 0; i < 10; ++i) {
        Worker worker ;
        worker.m_Name = "员工";
        worker.m_Name +=nameSeed[i];

        worker.m_Salary=rand()%10001+10000;  // 10000-20000
        //将员工放到容器中
        v.push_back(worker);

    }
}

//员工分组
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));
    }
}

void showWorkerByGroup(multimap<int,Worker> &m)
{
    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> vWorker;
    createWorker(vWorker);

//    for (vector<Worker>::iterator it = vWorker.begin();it != vWorker.end() ; it++)
//    {
//        cout << "姓名： " << it->m_Name << " 工资：" << it->m_Salary << endl;
//
//    }

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

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

    return 0;
}