最少猎头拿到全部简历问题

题目

在这里插入图片描述

已知猎头跟简历的对应关系,猎头手中的简历可能重叠。找到能获取全部简历的最少猎头

eg:
A 1 2 3 4
B 2 3 5
C 4 5 6
D 5 6 7 8
E 1 4 6
result: [A, D]

面试官:能做出来吗?
我:目前能想到的就是贪心算法(一直要简历最多的猎头,但是有可能拿到的不是最优解),可以用回溯解决吗?
面试官:你可以试一下

public class Solution {


    public static void main(String[] args) {

        Solution solution = new Solution();

        // 标记猎头及其拥有的简历
        HashMap<String, List<Integer>> listMap1 = new HashMap<>();

        listMap1.put("A", Arrays.asList(1, 2, 3, 4));
        listMap1.put("B", Arrays.asList(2, 3, 5));
        listMap1.put("C", Arrays.asList(4, 5, 6));
        listMap1.put("D", Arrays.asList(5, 6, 7, 8));
        listMap1.put("E", Arrays.asList(1, 4, 6));
        // [A, D]
        System.out.println(solution.query(listMap1));

        HashMap<String, List<Integer>> listMap = new HashMap<>();

        // 测试贪心算法错误的情况
        listMap.put("A", Arrays.asList(1, 3, 4, 5));
        listMap.put("B", Arrays.asList(1, 3, 6));
        listMap.put("C", Arrays.asList(2, 4, 5));
        listMap.put("D", Arrays.asList(1, 3, 6));
        // [C, D]
        System.out.println(solution.query(listMap));
    }


    public List<String> query(Map<String, List<Integer>> map){
        Set<Integer> set = new HashSet<>();
        List<String> nameList = new ArrayList<>();
        List<String> finalList = new ArrayList<>();
        Map<Integer, List<Integer>> nameNumMap = new LinkedHashMap<>();
        int num = 0;
        for (Map.Entry<String, List<Integer>> m : map.entrySet()) {
            set.addAll(m.getValue());
            nameList.add(m.getKey());
            nameNumMap.put(num, m.getValue());
            num++;
        }
        int[] visitor = new int[nameList.size()];
        int minTotalNum = nameList.size();
        dfs(0, set.size(), nameList.size(), nameList, nameNumMap, visitor, minTotalNum, finalList);
        return finalList;
    }

    /**
     * @param index 决定index个猎头是否选择
     * @param visitorNum 面试者总数
     * @param headNum 猎头总数
     * @param nameList 猎头名字
     * @param nameNumMap 数字->猎头拥有的简历
     * @param visit 标志是否取猎头的简历
     * @param minTotalNum 暂存获取全部简历的最少猎头数
     * @param finalList 最终选择的猎头名字
     */
    public void dfs(int index, int visitorNum, int headNum, List<String> nameList, Map<Integer, List<Integer>> nameNumMap,
                    int[] visit, int minTotalNum, List<String> finalList) {
        if (index > headNum) {
            return;
        }
        if (index == headNum) {
            // 选了几个猎头
            int total = 0;
            // 存放猎头的名字
            List<String> tempNameList = new ArrayList<>();
            // 拿到的简历
            Set<Integer> set = new HashSet<>();
            for (int i = 0; i < headNum; i++) {
                if (visit[i] == 1) {
                    total++;
                    tempNameList.add(nameList.get(i));
                    // 标记面试官中的简历
                    List<Integer> list = nameNumMap.get(i);
                    set.addAll(list);
                }
            }
            // 能拿到所有的简历
            if (set.size() == visitorNum) {
                if (total <= minTotalNum) {
                    finalList.clear();
                    finalList.addAll(tempNameList);
                }
            }
        } else {
            for (int i = 0; i < headNum; i++) {
                // 取第i个猎头
                visit[i] = 1;
                dfs(index + 1, visitorNum, headNum, nameList, nameNumMap, visit, minTotalNum, finalList);
                visit[i] = 0;
            }
        }
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值