【C++】学习笔记[十九]

STL【四】

1、STL案例练习----员工分组

案例描述

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

步骤实现

• 创建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 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-9999
		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));
	}
}

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 << "name:" << pos->second.m_name << "   " << "salary:" << pos->second.m_salary << endl;
	}
	cout << endl;
	cout << "***************************************" << endl;

	cout << "美术部门信息：" << endl;
	pos = m.find(MEISHU);
	count = m.count(MEISHU);//统计美术部门具体人数
	index = 0