【华为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;}
}