C++ STL案例2-员工分组

案例-员工分组

案例描述

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

实现步骤

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

示例：

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

#define CEHUA  1
#define MEISHU 2
#define YANFA  3
/*
-公司今天招聘了10个员工（ABCDEFGHIJ），10名员工进入公司后，需要指派员工到各个部门工作
- 员工信息有：姓名、工资组成；部门分为：策划、美术、研发
- 随机给10个员工分配部门和工资
- 通估multimap进行信息的插入， key（部门编号）value（员工）
- 分部门显示员工信息
*/
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 num = rand() % 3 + 1;

		//将员工插入到分组中
		m.insert(make_pair(num, *it));
	}
}

//按部门显示员工
void showByDept(multimap<int, Worker> &m)
{
	int count=0;
	multimap<int, Worker>::iterator pos;

	cout << "策划部：" << endl;
	pos = m.find(CEHUA);
	count = m.count(CEHUA);
	for (int i = 0; pos != m.end() && i < count; pos++, i++)
	{
		cout << pos->second.m_Name	<< " 薪资：" << pos->second.m_Salary << endl;
	}

	cout << "美术部：" << endl;
	pos = m.find(MEISHU);
	count = m.count(MEISHU);
	for (int i = 0; pos != m.end() && i < count; pos++, i++)
	{
		cout << pos->second.m_Name << " 薪资：" << pos->second.m_Salary << endl;
	}

	cout << "研发部：" << endl;
	pos = m.find(YANFA);
	count = m.count(YANFA);
	for (int i = 0; pos != m.end() && i < count; pos++, i++)
	{
		cout << pos->second.m_Name<< " 薪资：" << pos->second.m_Salary << endl;
	}

}

void main()
{
	//随机数种子
	srand((unsigned int)time(NULL));

	//1、 创建员工
	vector<Worker> vWorker;
	createWorker(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);
	//测试
	/*for (multimap<int, Worker>::iterator it = mWorker.begin(); it != mWorker.end(); it++)
	{
		cout << "部门编号：" << it->first << " " << it->second.m_Name << " 工资" << it->second.m_Salary << endl;
	}*/

	//3、按部门显示员工
	showByDept(mWorker);
}