容器(vector与multimap)案例 - 员工分组
案例描述:
- 公司招聘了10个员工(ABCEDEFGJHIJ),员工进入公司后,需要指派员工在哪个部门工作
- 员工信息有:姓名 工资;部门分为: 策划,美术,研发
- 随机给10名员工分配部门和工资
- 通过multimap进行信息插入 key(部门编号)value(员工)
- 分部门显示员工信息
实现步骤:
- 创建10名员工,放到vector中
- 遍历vector容器,取出每个员工,进行随机分组
- 分组后,将员工部门号作为key,具体工作为value,放到multimap容器中
- 分部门显示员工信息
源码:
#include <iostream>
#include <cstdlib>
using namespace std;
#include <map>
#include <string>
#include <vector>
#include <ctime>
#define CEHUA 0
#define MEISHU 1
#define YANFA 2
class Worker
{
public:
string m_Name;
int m_Salary;
};
//创建员工
void createWoeker(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()%10000 + 10000;
//放入容器
v.push_back(worker);
}
}
//分组
void setGroup(vector<Worker>&v, multimap<int,Worker>& m)
{
for(vector<Worker>::iterator it = v.begin(); it != v.end(); it++)
{
//部门编号 0 1 2
int deptId = rand() % 3;
//key 部门 value具体员工
m.insert(make_pair(deptId,*it));
}
}
//分组显示
void showWorkerByGourp(multimap<int,Worker>& m)
{
cout << "策划部门:" << endl;
multimap<int,Worker>::iterator pos = m.find(CEHUA);
int count = m.count(CEHUA);
for(int index = 0; pos !=m.end() && index < count; index++, pos++ )
{
cout << "姓名: " << pos->second.m_Name << " 工资:" << pos->second.m_Salary << endl;
}
cout << "------------------------" << endl;
cout << "美术部门:" << endl;
pos = m.find(MEISHU);
count = m.count(MEISHU);
for(int index = 0; pos !=m.end() && index < count; index++, pos++ )
{
cout << "姓名: " << pos->second.m_Name << " 工资:" << pos->second.m_Salary << endl;
}
cout << "------------------------" << endl;
cout << "研发部门:" << endl;
pos = m.find(YANFA);
count = m.count(YANFA);
for(int index = 0; pos !=m.end() && index < count; index++, pos++)
{
cout << "姓名: " << pos->second.m_Name << " 工资:" << pos->second.m_Salary << endl;
}
}
int main()
{
srand((unsigned int)time(NULL));
vector<Worker>vWorker;
//创建员工
createWoeker(vWorker);
//测试
// for(vector<Worker>::iterator it = vWorker.begin(); it != vWorker.end(); it++)
// {
// cout << "姓名: " << it->m_Name << " 工资: " << it->m_Salary << endl;
// }
//员工分组
multimap<int,Worker>mWorker;
setGroup(vWorker,mWorker);
//分组显示员工
showWorkerByGourp(mWorker);
system("pause");
return 0;
}