[黑马程序员课程记录]C++提高部分10

3.10 案例-员工分组

3.10.1 案例描述

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

3.10.2 实现步骤

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

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

class employee
{
public:
	employee(string name, int money)
	{
		this->name = name;
		this->money = money;
	}

	string name;
	int money;
};

void makeemployee(vector<employee> &v)
{
	string makename = "ABCDEFGHIJ";

	for (int i = 0; i < 10; i++)
	{
		string name = "员工";
		name += makename[i];
		int money = rand() % 10000 + 10000;
		employee e(name, money);
		v.push_back(e);
	}
}

void makegroup(vector<employee> &v, multimap<int, employee>&m)
{
	for (vector<employee>::iterator it = v.begin(); it != v.end(); it++)
	{
		//产生随机部门编号
		int deptid = rand() % 3;
		//将员工插入分组
		m.insert(make_pair(deptid, *it));
	}
}

void showgroup(multimap<int, employee>&m)
{
	cout << "策划部门" << endl;
	multimap<int, employee>::iterator pos = m.find(0);
	int count = m.count(0);
	int index=0;
	for (;pos!=m.end()&&index<count;pos++,index++)
	{
		cout << " 姓名： " << pos->second.name << " 工资： " << pos->second.money << endl;
	}

	cout << "美术部门" << endl;
	pos = m.find(1);
	count = m.count(1);
	index = 0;
	for (; pos != m.end() && index < count; pos++, index++) 
	{
		cout << " 姓名： " << pos->second.name << " 工资： " << pos->second.money << endl;
	}

	cout << "研发部门" << endl;
	pos = m.find(2);
	count = m.count(2);
	index = 0;
	for (; pos != m.end() && index < count; pos++, index++)
	{
		cout << " 姓名： " << pos->second.name << " 工资： " << pos->second.money << endl;
	}
}
int main()
{
	srand((unsigned int)time(NULL));

	//创建员工
	vector<employee> v;
	makeemployee(v);

	//测试
	//for (vector<employee>::iterator it = v.begin(); it != v.end(); it++)
	//{
	//	cout << "姓名： " << it->name << " 工资： " << it->money << endl;
	//}

	//分组
	multimap<int, employee>m;
	makegroup(v,m);

	//分组展示员工
	showgroup(m);

	system("pause");
	return 0;
}