C++STL容器案例-员工分组

C++STL容器案例-员工分组

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


//实现步骤
//创建10名员工，放到vector中
//遍历vector容器，取出每个员工，进行随机分组
//分组后，将员工部门编号作为key，具体员工作为value，放入到multimap容器中
//分部门显示员工信息
#include <iostream>
#include <map>
#include <vector>
#include <string>
#include <ctime>
using namespace std;

#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() % 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)
	{
		//产生随机部门编号
		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);
	//2.员工分组
	multimap<int, Worker>mworker;
	setGroup(vworker, mworker);

	//3.分组显示员工
	showWorkerByGroup(mworker);
	return 0;
}
//当数据以键值对形式存在， 考虑用map或者multimap

C++STL容器案例-员工分组