STL案例 - 员工分组

案例描述

 实现步骤

#include <algorithm> //算法
#include <ctime>
#include <iostream>
#include <map>
#include <string>
#include <vector>

#define CEHUA 0 //策划
#define MEISHU 1 //美术 不要加分号，注意
#define YANFA 2 //研发

using namespace std;

//案例：员工分组
class Worker {
public:
    string m_Name;
    int m_Salary;
};
void create_worker(vector<Worker>& v)
{
    //遍历创建十名员工
    string nameSeed = "ABCDEFGHIGK";
    for (int i = 0; i < 10; i++) {
        Worker worker;
        worker.m_Name = "员工";
        worker.m_Name += nameSeed[i];
        worker.m_Salary = rand() % 10000 + 10000; // salary是薪资，是一个随机数一万到一万九千九百九十九。
        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; //第一个位置是0 1 2 第二个位置是数据迭代器
    }
    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; //第一个位置是0 1 2 第二个位置是数据迭代器
    }
    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; //第一个位置是0 1 2 第二个位置是数据迭代器
    }
}

void test02()
{
    srand((unsigned int)time(NULL)); //设置随机数种子

    // 1.创建员工
    vector<Worker> vWorker;
    create_worker(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; // map容器,容器中需要一个pair数据
    setGroup(vWorker, mWorker);
    // 3.分组显示员工
    showWorkerByGroup(mWorker);
}

int main()
{
    test02();
}