题目:
金融证券行业超好的薪酬待遇,吸引了大批的求职者前往应聘,小东也不例外,准备应聘一家证券公司。面试官为考察她的数据分析、处理和编码能力,为她准备了以下问题。
股票交易中,委托是指股票交易者在证券公司买卖股票。每手委托包括一个委托单号i、价格pi、买入或卖出标记di及交易数量qi。
交易处理中,需要把同类业务(买入或卖出)中相同价格的所有委托合并起来,形成一个清单。清单的第一部分为按价格降序排列的合并后的卖出委托,紧随其后的是按相同顺序排列的买入合并委托。证券公司比较关心的是比较抢手的s条合并委托信息,需要得到买入及卖出最抢手的s条合并委托。对于买入委托,抢手的是指报价高的委托,而卖出委托中报价低的较为抢手。若买或卖的合并委托数小于s条,则全部列入清单中。
现在小东拿到的是n个委托,请你帮忙找出最抢手的s个合并委托。
输入
输入有若干组,每组的第一行为两个正整数n和s(1<=n<=1000,1<=s<=50),分别表示委托数和最抢手的清单数,接下来的n行为具体的委托信息,每行包含3部分,第一部分为一个字母‘B’或‘S’,表示买入或卖出,后两部分为两个整数p和q,表示报价和数量。任何卖出委托的报价都比买入委托的报价高。
样例输入:
6 2
B 10 3
S 50 2
S 40 1
S 50 6
B 20 4
B 25 10
输出
输出不超过2s行合并委托清单,格式与输入相同。
样例输出
S 50 8
S 40 1
B 25 10
B 20 4
这道题看的时候头都大了,其实方法就是提取关键字,虽然写了很多但是最重要的是理解输出想要啥,无论用什么手段只要能达到最后结果就OK。这是AC的基本目的。
关键1:
这题要求我们从字母B和S区分:
B表示买入
S表示卖出
关键2:
要求将p合并,如果有重复的p,则使其q值叠加起来。这里是不是已经想到用Map了???
关键3:
输入的s表示,输出的清单数量,如果s比我们算出的清单大,则把清单全输出即可。
好了,这里关键点说完了,我的脑洞一向都比较大,这里我抓住Entry就不放手了,因为把这个当做了突破口,最后真的一次就搞出来了,具体在代码里体现,其实做了很多麻烦的重复的操作,不过输出正确就OK,优化的事情以后慢慢做就好:直接粘代码了!
import java.util.*;
/**
* Created by AceCream on 2017/4/1.
*/
public class Main {
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
while(sc.hasNext()) {
int n = sc.nextInt();
int s = sc.nextInt();
Map<Integer, Integer> mapB = new TreeMap<>();
Map<Integer, Integer> mapS = new TreeMap<>();
for (int i = 0 ;i < n; i++){
String choose = sc.next();
int price = sc.nextInt();
int count = sc.nextInt();
if (choose.equals("B")) {
if (mapB.containsKey(price)) {
mapB.put(price, count + mapB.get(price));
} else {
mapB.put(price, count);
}
} else {
if (mapS.containsKey(price)) {
mapS.put(price, count + mapS.get(price));
} else {
mapS.put(price, count);
}
}
}
//这之前的代码也没什么难的,去掉重复的,然后更新Map的value值罢了
Set<Map.Entry<Integer, Integer>> entrySetB = mapB.entrySet();
Set<Map.Entry<Integer, Integer>> entrySetS = mapS.entrySet();
//后面主要是玩这两个List
List<Integer> listB = new ArrayList<>();
List<Integer> listS = new ArrayList<>();
for (Map.Entry entryB : entrySetB) {
listB.add((Integer) entryB.getKey());
}
Collections.sort(listB);
Collections.reverse(listB);
if (listB.size()>s){
listB = listB.subList(0,s);
}
for (Map.Entry entryS : entrySetS) {
listS.add((Integer) entryS.getKey());
}
Collections.sort(listS);
if (listS.size()>s){
listS = listS.subList(0,s);
}
Collections.reverse(listS);
for (int i : listS) {
System.out.println("S "+i+" "+mapS.get(i));
}
for (int i : listB) {
System.out.println("B "+i+" "+mapB.get(i));
}
}
}
}