21/02/25刷题记录Day7

面试题 16.02. 单词频率

设计一个方法，找出任意指定单词在一本书中的出现频率。

你的实现应该支持如下操作：

WordsFrequency(book)构造函数，参数为字符串数组构成的一本书
get(word)查询指定单词在书中出现的频率

思路

1.这个题的要点在于创造一个全局的hashmap，在初始化的时候将所有的单词都写进map中。
2.然后在get()中实现查询函数

完整代码

public class WordsFrequency {
	HashMap<String, Integer> hashMap;
	public WordsFrequency(String[] book) {
		hashMap = new HashMap<String, Integer>();
		for (String str : book) {
			if(!hashMap.containsKey(str)) {
				hashMap.put(str, 1);
			}else {
				hashMap.put(str, hashMap.get(str) + 1);
			}
		}
    }
    
    public int get(String word) {
    	if(!hashMap.containsKey(word)) {
    		return 0;
    	}
    	return hashMap.get(word);
    }
}

面试题 17.14. 最小K个数

设计一个算法，找出数组中最小的k个数。以任意顺序返回这k个数均可。

思路

这个题写法很多。排序，大根堆都是常用的

完整代码

public int[] smallestK(int[] arr, int k) {
        if (k == 0) {
            return new int[0];
        }
        PriorityQueue<Integer> heap = new PriorityQueue<Integer>(new Comparator<Integer>() {
        	@Override
        	public int compare(Integer o1, Integer o2) {
        		// TODO Auto-generated method stub
        		return o2 - o1;
        	}
		});
        for (int i = 0; i < k; ++i) {
            heap.offer(arr[i]);
        }
        for (int i = k; i < arr.length; ++i) {
            if (heap.peek() > arr[i]) {
            	heap.poll();
            	heap.offer(arr[i]);
            }
        }
        int[] ans  = new int[k];
        for (int i = 0; i < k; ++i) {
        	ans[i] = heap.poll();
        }
        return ans;

	}