【华为OD题库-091】选举拉票-java

题目

现在你要竞选一个县的县长。你去对每一个选民进行了调查。你已经知道每一个人要选的人是谁，以及要花多少钱才能让这个人选你。现在你想要花最少的钱使得你当上县长。你当选的条件是你的票数比任何一个其它候选人的多(严格的多，不能和他们中最多的相等)。请计算一下最少要花多少钱。
输入
单组测试数据。
第一行有一个整数(1≤n≤10^5)，表示这个县的选民数目。
接下来有n行，每一行有两个整数ai和bi(0<=10^5,0<=bi<=10^4)，表示第i个选民选的是第ai号候选人，想要让他选择自己就要花bi的钱。你是0号候选人(所以，如果一个选民选你的话ai就是0，这个时候bi也肯定是0)。
输出
输出一个整数表示花费的最少的钱。
示例1:
输入∶
5
1 2
1 2
1 2
2 1
0 0
输出:
3

思路

大致同【华为OD题库-041】人气最高的店铺-java

之前是使用的自定义对象存储的数据，现在直接存入list<int []>中。
之前输入5个店铺，那么店铺编号范围一定在：0~4。所以可以使用int[]数组存放每个店铺的票情况，现在不确定有多少个竞选者，使用map存放每个竞选者的的票情况

题解

package hwod;

import java.util.*;

public class VoteMagistrate {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int[][] nums = new int[n][2];
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < 2; j++) {
                nums[i][j] = sc.nextInt();
            }
        }
        System.out.println(voteMagistrate(nums));

    }

    private static Map<Integer, Integer> voteMap = new HashMap<>();
    private static int res=Integer.MAX_VALUE;

    private static int voteMagistrate(int[][] nums) {
        List<int[]> list = new ArrayList<>();
        for (int i = 0; i < nums.length; i++) {
            voteMap.put(nums[i][0], voteMap.getOrDefault(nums[i][0], 0) + 1);
            if (nums[i][0] != 0) list.add(nums[i]);
        }
        int maxIdx = 0;//获取票数最多的竞选人代号
        for (Integer k : voteMap.keySet()) {
            if (k != 0 && voteMap.get(k) >= voteMap.getOrDefault(maxIdx, 0)) {
                maxIdx = k;
            }
        }
        if (maxIdx == 0) return 0;
        list.sort((o1, o2) -> {
            if (o1[0] != o2[0]) return o1[0] - o2[0];
            return o1[1] - o2[1];
        });
        
        int[] used = new int[list.size()];
        dfs(list, 0, used, 0);
        return res;


    }

    private static void dfs(List<int[]> list, int start, int[] used, int price) {
        if (check(voteMap)) {
            res = Math.min(price, res);
            return;
        }
        for (int i = start; i < list.size(); i++) {
            int[] cur = list.get(i);
            if (i > 0 && list.get(i)[0] == list.get(i - 1)[0] && used[i - 1] == 0) continue;
            used[i] = 1;
            voteMap.put(cur[0], voteMap.get(cur[0]) - 1);
            voteMap.put(0, voteMap.getOrDefault(0, 0) + 1);
            dfs(list, i + 1, used, price + cur[1]);
            used[i] = 0;
            voteMap.put(0, voteMap.get(0) - 1);
            voteMap.put(cur[0], voteMap.get(cur[0]) + 1);
        }
    }

    private static boolean check(Map<Integer, Integer> voteMap) {
        //判断0号是不是唯一最大的票
        int target = voteMap.getOrDefault(0, 0);
        for (Integer key : voteMap.keySet()) {
            if (key != 0 && voteMap.get(key) >= target) {
                return false;
            }
        }
        return true;
    }
}

推荐

如果你对本系列的其他题目感兴趣，可以参考华为OD机试真题及题解（JAVA），查看当前专栏更新的所有题目。