c++学习笔记-提高篇-案例2-员工分组（vector/multimap）

一、案例描述

• 公司今天招聘10个员工（ABCDEFGHIJ）,10名员工进入公司后，需要指派员工在哪个部门工作
• 员工信息：姓名? 工资组成；部门分为：策划、美术、研发
• 随机给10名员工分配部门和工作
• 通过multimap进行信息插入,? key(部门编号) value(员工)
• 分部门显示员工信息

二、实现步骤

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

三、代码示例

#include<iostream>
using namespace std;

#include<map>
#include<vector>
#include<ctime>
/*
*案例描述： 
公司今天招聘10个员工（ABCDEFGHIJ）,10名员工进入公司后，需要指派员工在哪个部门工作
员工信息：姓名? 工资组成；部门分为：策划、美术、研发
随机给10名员工分配部门和工作
通过multimap进行信息插入,? key(部门编号) value(员工)
分部门显示员工信息
*/

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

class Worker
{
public:

	string m_Name;
	int m_Salary;
};

void creatWorker(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; //10000~19999

		//将员工放到容器中
		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;
	creatWorker(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);


	system("pause");
	return 0;
}