为了找到自己满意的工作,牛牛收集了每种工作的难度和报酬。牛牛选工作的标准是在难度不超过自身能力值的情况下,牛牛选择报酬最高的工作。在牛牛选定了自己的工作后,牛牛的小伙伴们来找牛牛帮忙选工作,牛牛依然使用自己的标准来帮助小伙伴们。牛牛的小伙伴太多了,于是他只好把这个任务交给了你。
输入描述:
每个输入包含一个测试用例。 每个测试用例的第一行包含两个正整数,分别表示工作的数量N(N<=100000)和小伙伴的数量M(M<=100000)。 接下来的N行每行包含两个正整数,分别表示该项工作的难度Di(Di<=1000000000)和报酬Pi(Pi<=1000000000)。 接下来的一行包含M个正整数,分别表示M个小伙伴的能力值Ai(Ai<=1000000000)。 保证不存在两项工作的报酬相同。
输出描述:
对于每个小伙伴,在单独的一行输出一个正整数表示他能得到的最高报酬。一个工作可以被多个人选择。
输入例子1:
3 3 1 100 10 1000 1000000000 1001 9 10 1000000000
输出例子1:
100 1000 1001
import java.util.Iterator;
import java.util.Map;
import java.util.Scanner;
import java.util.TreeMap;
public class Main2 {
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
TreeMap<Integer, Integer> treeMap = new TreeMap<Integer, Integer>();
int n = sc.nextInt();
int m = sc.nextInt();
int b[]=new int[m];
for (int i = 0; i < n; i++) {
Integer k = sc.nextInt();
Integer v = sc.nextInt();
treeMap.put(k, v);
}
for (int i = 0; i < m; i++) {
Integer a = sc.nextInt();
b[i]=a;
if (!treeMap.containsKey(a)) {
treeMap.put(a, 0);
}
}
int max = 0;
Iterator<Map.Entry<Integer, Integer>> it = treeMap.entrySet().iterator();
while (it.hasNext()) {
Map.Entry<Integer, Integer> entry = it.next();
if (entry.getValue() < max) {
treeMap.put(entry.getKey(), max);
} else {
max=entry.getValue();
}
}
for (int i = 0; i < b.length; i++) {
System.out.println(treeMap.get(b[i]));
}
}
}
思路:定义一个TreeMap,以便根据key(能力值或难度)找到对应的value(报酬),另外TreeMap是会对key值进行自然排序的;首先使用treeMap.put(k, v)添加 K 工作的难度和 V 报酬,然后添加小伙伴能力值为K,如果treeMap不存在K,则添加此K值,value设为0,用于TreeMap是会对key值进行自然排序的,所以只要使用迭代器Iterator按顺序遍历treeMap,把比当前K值小的最大value和当前value比较,当前value如果小的话就替换为大的value,最后使用treeMap.get(b[i])把对应小伙伴能力值的最高报酬打印出来。