leetcode:Top K Frequent Elements

原题：
Given a non-empty array of integers, return the k most frequent elements.
For example,
Given [1,1,1,2,2,3] and k = 2, return [1,2].

Note:

You may assume k is always valid, 1 ≤ k ≤ number of unique elements.Your algorithm's time complexity must be better than O(n log n), where n is the array's size.

题意是给定体一个非空的整形数组和整数k，返回出现频率最高的k个元素

分析：既然是元素与其出现频率的对应关系，很快想到map类。
思路是遍历一遍数组，记录元素和其出现次数，元素存入key，出现次数存入value。然后对value进行从大到小排序，取前k个key即为所求。
需要进行排序，很自然想到TreeMap,但需注意的是TreeMap是对map中的key进行排序，而我们是对value进行排序，所以需要自己重写排序方法。

Java实现：

import java.util.ArrayList;import java.util.Collections;import java.util.Comparator;import java.util.HashMap;import java.util.List;import java.util.Map;import java.util.Map.Entry;public class Solution {    public List<Integer> topKFrequent(int[] nums, int k) {        List<Integer> result =new ArrayList<Integer>();        if(nums.length==0 || nums.length<k) return result;        Map<Integer,Integer> map = new HashMap<Integer,Integer>();        for (int i = 0; i < nums.length; i++) {//将数组元素放入map,对应的value是其出现的次数            int temp = nums[i];            if(map.containsKey(temp))                map.put(temp, map.get(temp)+1);            else                map.put(temp, 1);        }        List<Map.Entry<Integer, Integer>> list = new ArrayList<Map.Entry<Integer,Integer>>(map.entrySet());//map集合包装成list，对map中的value进行排序        Collections.sort(list,new Comparator<Map.Entry<Integer, Integer>>(){            @Override            public int compare(Entry<Integer, Integer> o1,                    Entry<Integer, Integer> o2) {//对value从大到小排序                return o2.getValue()-o1.getValue();            }        });        for (int j=0;j<k;j++) {//取前k个            //System.out.println(list.get(j).getKey()+" : "+list.get(j).getValue());            result.add(list.get(j).getKey());        }        return result;    }}

Python实现：

class Solution(object):    def topKFrequent(self, nums, k):        """        :type nums: List[int]        :type k: int        :rtype: List[int]        """        dict={}        result=[]        for i in nums:            if dict.has_key(i):                dict[i]+=1            else:                dict[i]=1        sortDict=sorted(dict.iteritems(),key=lambda  d:d[1],reverse=True)#对value进行排序        for x in sortDict[:k]:#sortDict是一个list，每一个元素是排序后的元组，取前k个            result.append(x[0])#元组的第一个是dict中对应的key        return result