2021-07-17

利用vector和map容器进行对员工分组的案例

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

#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() % 10001 + 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)
{
	// 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++)
	{
		cout << "姓名：" << pos->second.m_Name << " 工资：" << pos->second.m_Salary << endl;

	}
	cout << "------------------------" << endl;

	pos = m.find(MEISHU);
	count = m.count(MEISHU); //统计策划部门的具体人数
	index = 0;
	for (; pos != m.end() && index++ < count; pos++)
	{
		cout << "姓名：" << pos->second.m_Name << " 工资：" << pos->second.m_Salary << endl;

	}
	cout << "------------------------" << endl;

	pos = m.find(YANFA);
	count = m.count(YANFA); //统计策划部门的具体人数
	index = 0;
	for (; pos != m.end() && index++ < count; pos++)
	{
		cout << "姓名：" << pos->second.m_Name << " 工资：" << pos->second.m_Salary << endl;

	}

}

int main()
{
	srand((unsigned int)time(NULL));
	vector<Worker>vworker;
	CreateWorker(vworker);

	multimap<int, Worker>mworker;
	setGroup(vworker, mworker);
	showWorkerByGroup(mworker);

	return 0;

}