leetcode 347 priority，map的使用

主要是对次数进行排序，然后去前几个最大次数的值，输出即可

 class Solution {

 public:

     vector<int> topKFrequent(vector<int>& nums, int k) {

         int i;

         unordered_map<int,int>p_map;

         map<int,int>::iterator it;

         for(i=;i<nums.size();i++) {

             p_map[nums[i]]++;

         }

         priority_queue<pair<int,int>>p_queue;

         for(it=p_map.begin();it!=p_map.end();it++)

          p_queue.push(make_pair(it->second,it->first));

         vector<int>num_result;

         for(i=;i<=k;i++) {

             num_result.push_back(p_queue.top().second);

             p_queue.pop();

         }

         return num_result;

     }

 };
另一种写法也非常好：http://blog.csdn.net/yzhang6_10/article/details/51388021

 class Op{

     public:

      int Op_count;

      int Op_num;

      Op(int count,int num) {

          Op_count=count;

          Op_num=num;

      }

      bool operator <(const Op&Cpone)const

      {

          return Op_count>Cpone.Op_count;

      }

 };

 class Solution {

 public:

     vector<int> topKFrequent(vector<int>& nums, int k) {

         int i;

         map<int,int>p_map;

         map<int,int>::iterator it;

         for(i=;i<nums.size();i++) {

             p_map[nums[i]]++;

         }

         vector<Op>p_queue;

         for(it=p_map.begin();it!=p_map.end();it++)

          p_queue.push_back(Op(it->second,it->first));

         vector<int>num_result;

         sort(p_queue.begin(),p_queue.end());

         for(i=;i<k;i++)

             num_result.push_back(p_queue[i].Op_num);

         return num_result;

     }

 };

巴特西

leetcode 347 priority，map的使用

最新文章

热门文章