标签：结构 eset 实例 abc stat off .com 复杂 cat
第二题：有一个数据流
持续的读入以下数据：stock，price.
每天早上都是从empty开始读入数据
设计一个application，能始终返回股票价格变化程度top k的股票
我用的hashmap加priorityqueue做的，每次有新数据的时候，检查hashmap里面有没有，如果没有的话，插入到hashmap和priorityqueue。如果有的话，删除priorityqueue里面的东西，然后更新price diff再插入，每次top k的request都是pop出k的然后再插回去。
这轮我不知道maxheap还有一个findnext的功能（这是面试官说的，虽然我当时感觉maxheap应该不可能实现constant time findnext功能的，并且给他解释了为什么，但是他还是说有这样一个method，但是我doubt他知道这个method的内部实现。当然我也没有继续坚持，只是说谢谢你让我知道了一个新method。后来我查了java的priorityqueue的methods里并没有这样一个iterator，希望有知道的同学可以讨论一下），所以每次request的时候都是先pop出来再插进去导致了klgn的复杂度，最后面试官说有这样一个method能够constant time返回下一个大的node, 这样top k request的时候就是O(k)的复杂度。
import java.util.Comparator;
import java.util.HashMap;
import java.util.PriorityQueue;
class Solution {
private class Node {
String name;
double diff;
double pri;
public Node(String name, double diff, double pri) {
this.name = name;
this.diff = diff;
this.pri = pri;
}
}
HashMap<String, Node> map = new HashMap<>();
Comparator<Node> com = new Comparator<Node>() {
@Override
public int compare(Node o1, Node o2) {
return (int)(o1.diff - o2.diff);
}
};
PriorityQueue<Node> pq = new PriorityQueue<>(com);
public void topK(String[][] prices, int k) {
for (String[] cur : prices) {
if (!map.containsKey(cur[0])) {
map.put(cur[0], new Node(cur[0], 0.0, Double.valueOf(cur[1])));
} else {
Node pre = map.get(cur[0]);
double dif = Math.abs(pre.pri - Double.valueOf(cur[1]));
if (pq.contains(pre)) {
pq.remove(pre);
}
Node newNode = new Node(cur[0], dif, Double.valueOf(cur[1]));
map.put(cur[0], newNode);
pq.add(newNode);
while (pq.size() > k) {
pq.poll();
}
}
}
for (Node no : pq) {
System.out.println(no.name + no.pri);
}
}
public static void main(String[] args) {
// TODO Auto-generated method stub
Solution sol = new Solution();
//String[][] nums = new String[][]{{"aa", "1.1"}, {"aa", "2.5"}, {"b", "1"}, {"aa", "2.9"}};
//int[][] lands = new int[][]{{1, 1, 0}, {1, 1, 0}, {1, 1, 0}};
//String s = "abcacd";
//System.out.println(sol.sort(lands));
String[][] nums = new String[][]{{"b", "1.1"}, {"aa", "2.5"}, {"b", "1"}, {"aa", "2.9"}};
sol.topK(nums, 2);
}
TreeMap 或者 TreeSet 删除又可以 o(lgn), 插入也是O(lgn)。 这样就快了。 under the hood, 其实就是一个BST.
hashmap用以快速更新 node 的max, min value, 然后treeset 就可以自动lgn调整 key (diff) 的顺序了。
import java.util.Comparator;
import java.util.HashMap;
import java.util.TreeSet;
class Solution {
private class Node {
String name;
double diff;
double pri;
public Node(String name, double diff, double pri) {
this.name = name;
this.diff = diff;
this.pri = pri;
}
}
HashMap<String, Node> map = new HashMap<>();
Comparator<Node> com = new Comparator<Node>() {
@Override
public int compare(Node o1, Node o2) {
return (int)(o1.diff - o2.diff);
}
};
TreeSet<Node> pq = new TreeSet<>(com);
public void topK(String[][] prices, int k) {
for (String[] cur : prices) {
if (!map.containsKey(cur[0])) {
map.put(cur[0], new Node(cur[0], 0.0, Double.valueOf(cur[1])));
} else {
Node pre = map.get(cur[0]);
double dif = Math.abs(pre.pri - Double.valueOf(cur[1]));
if (pq.contains(pre)) {
pq.remove(pre);
}
Node newNode = new Node(cur[0], dif, Double.valueOf(cur[1]));
map.put(cur[0], newNode);
pq.add(newNode);
System.out.println(pq.size());
System.out.println(pq.first().name + pq.first().pri + " " + pq.last().name + " " + pq.last().pri);
while (pq.size() > k) {
pq.pollFirst();
}
}
}
for (Node no : pq) {
System.out.println(no.name + no.pri);
}
}
public static void main(String[] args) {
// TODO Auto-generated method stub
Solution sol = new Solution();
String[][] nums = new String[][]{{"aa", "1.1"}, {"bb", "1"}, {"bb", "1.8"},{"aa", "2.5"}, {"aa", "2.51"}};
sol.topK(nums, 2);
}
}
第二轮：. more info on
第一题：leetcode 232
第二题：一个matrix里，有一些element有treasure，从左上角走到右下角，只能往下往右，求最大能得到的treasure的值，标准二维dp解的
第三题：leetcode 138, 我说我做过了，他就换了一道
第四题：leetcode 98
标签：结构 eset 实例 abc stat off .com 复杂 cat
原文地址：http://www.cnblogs.com/apanda009/p/7855788.html
