forked from kamyu104/LeetCode-Solutions
-
Notifications
You must be signed in to change notification settings - Fork 14
/
smallest-sufficient-team.cpp
36 lines (35 loc) · 1.25 KB
/
smallest-sufficient-team.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
// Time: O(m * 2^n), n is the number of skills
// m is the number of people
// Space: O(2^n)
class Solution {
public:
vector<int> smallestSufficientTeam(vector<string>& req_skills, vector<vector<string>>& people) {
unordered_map<string, int> lookup;
for (int i = 0; i < req_skills.size(); ++i) {
lookup[req_skills[i]] = i;
}
unordered_map<int, vector<int>> dp;
dp[0];
for (int i = 0; i < people.size(); ++ i) {
int his_skill_set = 0;
for (const auto& skill : people[i]) {
if (lookup.count(skill)) {
his_skill_set |= 1 << lookup[skill];
}
}
auto tmp(dp);
for (const auto& [skill_set, people] : tmp) {
const auto with_him_set = skill_set | his_skill_set;
if (with_him_set == skill_set) {
continue;
}
if (!dp.count(with_him_set) ||
dp[with_him_set].size() > people.size() + 1) {
dp[with_him_set] = move(people);
dp[with_him_set].emplace_back(i);
}
}
}
return dp[(1 << req_skills.size()) - 1];
}
};