现在有n位工程师和6项工作(编号为0至5),现在给出每个人能够胜任的工作序号表(用一个字符串表示,比如:045,表示某位工程师能够胜任0号,4号,5号工作)。现在需要进行工作安排,每位工程师只能被安排到自己能够胜任的工作当中去,两位工程师不能安排到同一项工作当中去。如果两种工作安排中有一个人被安排在的工作序号不一样就被视为不同的工作安排,现在需要计算出有多少种不同工作安排计划。
输入数据有n+1行: 第一行为工程师人数n(1 ≤ n ≤ 6) 接下来的n行,每行一个字符串表示第i(1 ≤ i ≤ n)个人能够胜任的工作(字符串不一定等长的)
输出一个整数,表示有多少种不同的工作安排方案
本题可采用回溯算法,代码如下:
#include<iostream>
#include<string>
#include<unordered_map>
#include<vector>
using namespace std;
void backTrack(vector<string> arr, int &count, int index, unordered_map<char, int> m);
int main()
{
int n;
cin >> n;
vector<string> arr;
string tmp;
for (int i = 0; i<n; i++)
{
cin >> tmp;
arr.push_back(tmp);
}
int count = 0;
unordered_map<char, int> m;
backTrack(arr, count, 0, m);
cout << count;
return 0;
}
void backTrack(vector<string> arr, int &count, int index, unordered_map<char, int> m) {
if (index == arr.size()) //满足条件,输出结果
{
count++;
return;
}
string str = arr[index];
for (int i = 0; i<str.size(); i++) //枚举所有可能的路径
{
if (m[str[i]] == 0) //满足界限函数和约束条件
{
m[str[i]] = 1;
backTrack(arr, count, index + 1, m);
m[str[i]] = 0;//回溯清理工作
}
}
}