STL案例-员工分组

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

#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;

		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;

		//将员工插入到分组中。key部门编号，value具体员工
		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() {
	vector<Worker>v;
	createWorker(v);

	multimap<int, Worker>m;
	setGroup(v,m);
	showWorkerByGroup(m);
}