【算法题】工作安排

来源:互联网 发布:雪橇三傻 知乎 编辑:程序博客网 时间:2024/05/16 02:38

现在有n位工程师和6项工作(编号为0至5),现在给出每个人能够胜任的工作序号表(用一个字符串表示,比如:045,表示某位工程师能够胜任0号,4号,5号工作)。现在需要进行工作安排,每位工程师只能被安排到自己能够胜任的工作当中去,两位工程师不能安排到同一项工作当中去。如果两种工作安排中有一个人被安排在的工作序号不一样就被视为不同的工作安排,现在需要计算出有多少种不同工作安排计划。
输入描述:
输入数据有n+1行:
第一行为工程师人数n(1 ≤ n ≤ 6)
接下来的n行,每行一个字符串表示第i(1 ≤ i ≤ n)个人能够胜任的工作(字符串不一定等长的)

输出描述:
输出一个整数,表示有多少种不同的工作安排方案

输入例子:
6
012345
012345
012345
012345
012345
012345

输出例子:
720

注意题意的理解:
1、一个人只能做一项工程,而不能分饰两角;
2、有几个工程师,每个工程师有一个工作即满足题意,不用6项工作全部都要有人做。

#include <iostream>#include <vector>#include <string>#include <algorithm>using namespace std;int func(vector<vector<int>>& vec,int deep,vector<int>& set_){    if (deep == vec.size())    {        return 1;    }    int count(0);    for (auto j = 0; j < 6;++j)    {        if (vec[deep][j]!=0&&  (set_[j]==0)   )        {            set_[j] = 1;            count += func(vec, deep + 1, set_);            set_[j] = 0;        }    }    return count;}int main(){    int n;    cin >> n;    vector<string> vec;    vec.resize(n);    for (auto i = 0; i < n;++i)    {        cin >> vec[i];    }    vector<vector<int>> vec_;    vec_.resize(n);    for (auto i = 0; i < n; ++i)    {        vec_[i].resize(6);        for (auto j :vec[i])        {            int index = j - '0';            vec_[i][index] = 1;        }    }    vector<int> set_(6,0);    cout<<func(vec_,0,set_);    return 0;}