1125. 最小的必要团队

作为项目经理,你规划了一份需求的技能清单 req_skills,并打算从备选人员名单 people 中选出些人组成一个「必要团队」( 编号为 i 的备选人员 people[i] 含有一份该备选人员掌握的技能列表)。

所谓「必要团队」,就是在这个团队中,对于所需求的技能列表 req_skills 中列出的每项技能,团队中至少有一名成员已经掌握。可以用每个人的编号来表示团队中的成员:

例如,团队 team = [0, 1, 3] 表示掌握技能分别为 people[0],people[1],和 people[3] 的备选人员。
请你返回 任一 规模最小的必要团队,团队成员用人员编号表示。你可以按 任意顺序 返回答案,题目数据保证答案存在。

示例 1:

输入:req_skills = ["java","nodejs","reactjs"], people = [["java"],["nodejs"],["nodejs","reactjs"]]
输出:[0,2]
示例 2:

输入:req_skills = ["algorithms","math","java","reactjs","csharp","aws"], people = [["algorithms","math","java"],["algorithms","math","reactjs"],["java","csharp","aws"],["reactjs","csharp"],["csharp","math"],["aws","java"]]
输出:[1,2]
 

提示:

1 <= req_skills.length <= 16
1 <= req_skills[i].length <= 16
req_skills[i] 由小写英文字母组成
req_skills 中的所有字符串 互不相同
1 <= people.length <= 60
0 <= people[i].length <= 16
1 <= people[i][j].length <= 16
people[i][j] 由小写英文字母组成
people[i] 中的所有字符串 互不相同
people[i] 中的每个技能是 req_skills 中的技能
题目数据保证「必要团队」一定存在

来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/smallest-sufficient-team
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

状态压缩DP+pre数组记录路径

如果只计算最小数量的话,那么这道题和 691. 贴纸拼词 差不多;但还得记录最小数量的路径,pre数组不好搞;肝了一上午,终于肝出来了;不容易啊,记录一下。

预处理req_skills和people数组,将req_skills数组内的技能编号,将people内的数组转换为int数,每个人的技能列表都可以转换为一个int数,这个int数的第i位为1表示掌握第i个技能。那么就可以使用位运算来动态规划了。

递推公式为: f[i | peo[j]] = min(f[i | peo[j]], f[i] + 1)
使用二维数组pre来记录路径,表示第i个状态选用第pre[i][1]个人,且前一个状态的下标为pre[i][0]

作者:lanrengufeng
链接:https://leetcode.cn/problems/smallest-sufficient-team/solution/d-by-lanrengufeng-1z9x/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

class Solution {
    public int[] smallestSufficientTeam(String[] req_skills, List<List<String>> people) {
        int n = req_skills.length;
        int[] peo = new int[people.size()];
        int[] f = new int[1 << n];
        int[][] pre = new int[1 << n][2];
        Arrays.fill(f, Integer.MAX_VALUE / 2);
        f[0] = 0;
        Map<String, Integer> map = new HashMap<>();
        for (int i = 0; i < n; i++) {
            map.put(req_skills[i], i);
        }
        for (int i = 0; i < people.size(); i++) {
            int cur = 0;
            for (int j = 0; j < people.get(i).size(); j++) {
                cur |= 1 << map.get(people.get(i).get(j));
            }
            peo[i] = cur;
        }
        for (int i = 0; i < f.length; i++) {
            for (int j = 0; j < peo.length; j++) {
                if (f[i] + 1 < f[i | peo[j]]) {
                    f[i | peo[j]] = f[i] + 1;
                    pre[i | peo[j]][0] = i;
                    pre[i | peo[j]][1] = j;
                }
            }
        }

        int[] ans = new int[f[f.length - 1]];
        for (int i = 0, idx = f.length - 1; i < ans.length; i++) {
            ans[i] = pre[idx][1];
            idx = pre[idx][0];
        }

        return ans;
    }
}

作者:lanrengufeng
链接:https://leetcode.cn/problems/smallest-sufficient-team/solution/d-by-lanrengufeng-1z9x/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
  • 3
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值