黑马程序员——STL员工分组

#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 CreateWoker(vector<Worker>& v) {
	string seed = "ABCDEFGHIJ";

	for (int i = 0; i < 10; i++) {
		Worker w;
		w.m_Name = "员工";
		w.m_Name += seed[i];
		w.m_Salary = rand() % 10000 + 10000;

		v.push_back(w);
	}
}

void SetGroup(vector<Worker>& v, multimap<int, Worker>& m) {
	for (vector<Worker>::iterator it = v.begin(); it != v.end(); it++) {
		//产生部门编号
		int depId = rand() % 3;
		m.insert(make_pair(depId, *it));
	}
}


void ShowWoker(multimap<int, Worker>& m) {
	cout << "策划部门：" << endl;
	multimap<int, Worker>::iterator it = m.find(cehua);
	int count = m.count(cehua);
	for (int i = 0; i < count; i++) {
		cout << "姓名；" << it->second.m_Name << " 工资：" << it->second.m_Salary << endl;
		it++;
	}

	cout << "--------------------------------------" << endl;
	cout << "美术部门：" << endl;
	it = m.find(meishu);
	count = m.count(meishu);
	for (int i = 0; i < count; i++) {
		cout << "姓名；" << it->second.m_Name << " 工资：" << it->second.m_Salary << endl;
		it++;
	}

	cout << "--------------------------------------" << endl;
	cout << "研发部门：" << endl;
	it = m.find(yanfa);
	count = m.count(yanfa);
	for (int i = 0; i < count; i++) {
		cout << "姓名；" << it->second.m_Name << " 工资：" << it->second.m_Salary << endl;
		it++;
	}
}

int main() {
	srand((unsigned int)time(NULL));

	vector<Worker>vWorker;
	CreateWoker(vWorker);
	
	multimap<int, Worker>mw;
	SetGroup(vWorker, mw);
	
	ShowWoker(mw);

	cout << endl;

	system("pause");

	return 0;
}