STL案例:员工分类

/************************************************************************/
/* 公司招聘了10名员工(ABCDEFGHI),员工进入公司后需要指派到哪个部门工作
/* 员工信息:姓名 工资组成
/* 部门分类:策划、美术、研发
/* 随机给10名员工分配部门和工资
/* 通过multimap进行信息的插入 key(部门编号) value(员工)
/* 分部门显示员工信息
/*
/* 实现
/* 1. 创建10名员工,放入vector中
/* 2. 遍历vector容器,取出每个员工,进行随机分组
/* 3. 分组后,将员工部门编号作为key,具体员工作为value,放入multimap容器
/* 4. 分部门显示员工信息
/************************************************************************/

#include 
#include 
#include 
#include 
#include 
#include 

using namespace std;

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

class Worker
{
public:
    string m_Name;
    int m_Salary;
};

void creatWorker(vector& worker)
{
    string name = "ABCDEFGHIJ";
    for (int i = 0; i < 10; i++)
    {
        Worker wo;
        wo.m_Name = "员工";
        wo.m_Name += name[i];
        wo.m_Salary = rand() % 10000 + 10000;
        worker.push_back(wo);
    }
}
void printWorker(const vector& v)
{
    for (vector::const_iterator it = v.begin(); it != v.end(); it++)
    {
        cout << "姓名:" << it->m_Name << "      薪水:" << it->m_Salary << endl;
    }

}
void setGroup(vector&v, multimap&m)
{
    for (vector::iterator it = v.begin(); it != v.end(); it++)
    {
        //随机的部门编号
        int deptId = rand() % 3;
        m.insert(make_pair(deptId, (*it)));
    }

}

void showWorkerByGroup(multimap& m)
{
    cout << "策划部门: " << endl;
    //查找策划部门员工的起始位置,返回迭代器
    multimap::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 << endl;
    }
}

int main()
{
    //随机数种子
    srand((unsigned int)time(NULL));
    // 创建员工
    vector vWorker;
    creatWorker(vWorker);
    //printWorker(vWorker);
    //员工分组
    multimap mWorker;
    setGroup(vWorker, mWorker);
    //分组显示员工
    showWorkerByGroup(mWorker);
    system("pause");
    return 0;
}

你可能感兴趣的:(STL案例:员工分类)