2021-8-20 C++提升学习 -＞ 3.10、STL案例2--员工分组

#include <iostream>
#include <string>
#include <map>
#include <vector>
#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 = "ABCDEFGHIJK";
	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;

		//将员工插入到分组中
		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;
	creatWorker(vWorker);

	//员工分组
	multimap<int, Worker>mWorker;
	setGroup(vWorker,mWorker);

	//分组显示员工
	showWorkerByGroup(mWorker);

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

	system("pause");
}