STL案例—员工分组

STL案例-员工分组

1 案例描述

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

2 实现步骤

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

案例代码：

#include <iostream>
#include <vector>
#include <map>
#include <string>
#include <cmath>
#include <time.h>
using namespace std;

#define CHEHUA 0
#define MEISHU 1
#define YANFA  2



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

//实现步骤

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

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;
		string name = "员工";
		name += nameSeed[i];
		worker.m_Name = name;
		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 随机数分组
		//员工分组
		m.insert(make_pair(deptId, *it));
	}
}

//4. 分部门显示员工信息
void showWorkerMessage(multimap<int,Worker> &m)
{
	cout << "策划部门： " << endl;
	multimap<int, Worker>::iterator pos = m.find(CHEHUA);//返回的是迭代器;
	int count = m.count(CHEHUA);//记录策划部门人数总数
	int index = 0;
	for (; pos != m.end() && index < count; pos++, index++)
	{
		cout << "姓名： " << pos->second.m_Name << "  工资： " << pos->second.m_Salary << 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;
	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. 创建10名员工，放到vector中
	vector<Worker> vWorker;
	createWorker(vWorker);

	//2. 遍历vector容器，取出每个员工，进行随机分组
	//3. 分组后，将员工部门编号作为key，具体员工作为value，放入到multimap容器中
	multimap<int, Worker> mWorker;
	setGroup(vWorker,mWorker);

	//4. 分部门显示员工信息
	showWorkerMessage(mWorker);

	//测试
	//for (vector<Worker>::iterator it = vWorker.begin(); it != vWorker.end(); it++)
	//{
	//	cout << "员工姓名： " << it->m_Name << "  员工工资： " << it->m_Salary << endl;
	//}

	system("pause");

	return 0;
}

总结：

• 当数据以键值对形式存在，可以考虑用map 或 multimap