STL案例2-员工分组

需求

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

#pragma once//防止头文件包含
#include<string>
#include<fstream>
#include <iostream>
#include<algorithm>//标准算法头文件
#include<vector>
#include<deque>
#include<ctime>
#include<stack>
#include<queue>
#include<list>
#include<set>
#include<map>
#include"myArray.hpp"
using namespace std;

/*
公司招聘10个员工（ABCDEFGHIJ）,10名员工进入公司后，需要指派员工在哪个部门工作
员工信息有：姓名  工资组成  ，部门：策划、美术、研发
随机给10名员工分配部门和工资
通过multimap进行信息的插入  key（部门编号）value（员工）、
分部门显示员工信息
*/

#define CEHUA 0
#define MEISHU 1
#define YANFA 2

class Worker// 员工类
{
public:
	string m_name;
	int m_salary;
};

void createWorker(vector<Worker>&v)//创建员工vector，并初始化员工信息
{
	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;//0 1 2

		//将员工插入到分组中
		//key部门编号，value具体员工
		m.insert(make_pair(deptId, *it));
	}
}


//员工显示
void showWorkByGourp(multimap<int, Worker>&m)
{
	//0 A B C  1 D E 2 F G..
	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;
	multimap<int, Worker>::iterator pos1 = m.find(MEISHU);
	int count1 = m.count(MEISHU);//统计策划部门具体人数
	int index1 = 0;
	for (;pos1 != m.end() && index1 < count1;pos1++, index1++)
	{
		cout << pos1->second.m_name << " " << pos1->second.m_salary << endl;
	}
	cout << "---------" << endl;
	cout << "研发部门：" << endl;
	multimap<int, Worker>::iterator pos2 = m.find(YANFA);
	int count2 = m.count(YANFA);//统计策划部门具体人数
	int index2 = 0;
	for (;pos2 != m.end() && index2 < count2;pos2++, index2++)
	{
		cout << pos2->second.m_name << " " << pos2->second.m_salary << endl;
	}
}

int main()
{
	//1创建员工
	vector<Worker>vWorker;
	createWorker(vWorker);

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

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

	//测试
	//for (vector<Worker>::iterator it = vWorker.begin();it != vWorker.end();it++)
	//{
	//	cout << it->m_name << " " << it->m_salary << endl;
	//}
	system("pause");
	return 0;
}

最后结果