STL案例--员工分组

1.案例描述

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

2.实现步骤

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

#include<iostream>
using namespace std;
#include<vector>
#include<map>

#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;//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 depId = rand() % 3;//0 1 2 

		//将员工插入分组
		m.insert(make_pair(depId, *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;
	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;

	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()
{
	//1.创建员工
	vector<Worker>v;
	createWorker(v);
	
	/*for (vector<Worker>::iterator it = v.begin(); it != v.end(); it++)
	{
		cout << "员工姓名：" << it->m_Name << " 工资：" << it->m_Salary << endl;
	}*/

	//2.员工分组
	multimap<int, Worker>m;//int-部门编号，Worker-具体员工
	setGroup(v, m);

	//3.分组显示员工
	showWorkerByGroup(m);

	system("pause");
	return 0;
}