员工分组-STL案例

案例描述：
1、公司今天招聘了10个员（ABCDEFGHUIJ），10名员工进入公司后，需要指派员工在那个部门工作。
2、员工信息有：姓名 工资组成；部门分为：策划、美术、研发
3、随机给10名员工分配部门和工资
4、通过multimap进行信息的插入 key（部门编号）
value（员工）
5、分部门显示员工信息

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

源代码：
#include
#include
#include

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

using namespace std;
class Worker
{
public:

string m_Name;
int m_Salary;

};

void createWorker(vector& 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;//10000~19999
	//将员工放入容器中
	v.push_back(worker);
}

}

//员工分组
void setGroup(vector& v, multimap<int, Worker>& m)
{
for (vector::iterator it = v.begin(); it != v.end(); it++)
{
//产生随机部门编号
int deptID = rand() % 3;//0,1,2

	//将员工插入到分组中
	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++, 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()
{
srand((unsigned int)time(NULL));

//1、创建员工
vector<Worker> vWorker;
createWorker(vWorker);

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

//3、分组显示员工
showWorkerByGroup(mWorker);
测试
//for (vector<Worker>::iterator it = vWorker.begin(); it != vWorker.end(); it++)
//{
//	cout << "姓名：" << it->m_Name << " 工资：" << it->m_Salary << endl;
//}

}
这个案例强化了我对map中的count函数和find函数 的使用，熟练掌握stl的用法可以提高编程效率。