C++程序例：员工分组案例 STL之multimap的小练习

#include<iostream>
#include <vector>
#include <map>
#include <time.h>

#define CEHUA 0
#define MEISHU 1
#define YANFA 2
using namespace std;

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 w;
		w.m_name = "员工";
		w.m_name += nameSeed[i];
		w.m_salary = rand() % 10000 + 10000;//10000~19999

		v.push_back(w);
	}
}

void init_test(vector<Worker>&v)
{
	cout << "显示新招的10名员工及薪资：" << endl;
	for (vector<Worker>::iterator it = v.begin(); it != v.end(); it++)
	{
		cout << it->m_name << " " << (*it).m_salary << endl;
	}
	cout << endl;
}

void setGroup(vector<Worker>&v,multimap<int,Worker>&mm)
{
	for (vector<Worker>::iterator it = v.begin(); it != v.end(); it++)
	{
		//对每个员工产生一个0、1、2之间的随机的部门编号
		int deptId = rand() % 3;
		//把员工的部门和员工类型的对象（员工自己）作为键值对插入到新建的map中
		mm.insert(pair<int, Worker>(deptId, *it));	//或mm.insert(make_pair(deptId, *it));
	}
}

void showWorkerByGroup(multimap<int, Worker>mm)
{
	cout << "显示10名员工的归组情况：" << endl<<endl;
	cout << "****** 策划部门 ******" << endl;
	multimap<int, Worker>::iterator pos = mm.find(CEHUA);//.find()找到第一个策划部门的员工的迭代器
	
	int count = mm.count(CEHUA);//统计策划部门分到的人数
	int index = 0;
	for (; pos != mm.end()&&index<count; pos++,index++)//map中按键排序的，所以pos++就会到下一个策划部的人
	{
		cout << "姓名：" << pos->second.m_name << " 薪资：" << pos->second.m_salary << endl;
	}
	cout << "——————————————————" << endl;
	cout << "****** 美术部门 ******" << endl;
	pos = mm.find(MEISHU);//.pos/count/index上面已经声明了，不需要再声明，直接复用变量即可，否则会出现多次初始化的报错
	count = mm.count(MEISHU);
	index = 0;
	for (; pos != mm.end() && index < count; pos++, index++)
	{
		cout << "姓名：" << pos->second.m_name << " 薪资：" << pos->second.m_salary << endl;
	}
	cout << "——————————————————" << endl;
	cout << "****** 研发部门 ******" << endl;
	pos = mm.find(YANFA);//.find()找到第一个策划部门的员工的迭代器
	count = mm.count(YANFA);//统计策划部门分到的人数
	index = 0;
	for (; pos != mm.end() && index < count; pos++, index++)
	{
		cout << "姓名：" << pos->second.m_name << " 薪资：" << pos->second.m_salary << endl;
	}
}


int main()
{
	srand((unsigned)time(NULL));//随机数种子
	vector<Worker>v;//创建vector放员工
	createWorker(v);//实例化10个worker对象并放入创建的vector中
	init_test(v);//显示一下新招的10名员工及薪资
	multimap<int, Worker>mm;//创建一个map
	setGroup(v,mm);//把员工分组存入创建的map中
	showWorkerByGroup(mm);//分组显示员工，此时所有信息都存在了map中，所以只需要传入mm
	system("pause");
	return 0;
}

显示新招的10名员工及薪资：
员工A 13194
员工B 18237
员工C 13538
员工D 12603
员工E 14900
员工F 19942
员工G 13531
员工H 12235
员工I 13855
员工J 16700

显示10名员工的归组情况：

****** 策划部门 ******
姓名：员工C 薪资：13538
姓名：员工D 薪资：12603
姓名：员工I 薪资：13855
姓名：员工J 薪资：16700
——————————————————
****** 美术部门 ******
姓名：员工B 薪资：18237
姓名：员工F 薪资：19942
——————————————————
****** 研发部门 ******
姓名：员工A 薪资：13194
姓名：员工E 薪资：14900
姓名：员工G 薪资：13531
姓名：员工H 薪资：12235
请按任意键继续. . .