案例描述
实现步骤
#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();
}