347. 前 K 個高頻元素(STL)

  • 2020 年 2 月 17 日
  • 筆記

       雖然tag是中等難度,感覺還是比較簡單的,不難想到對每一個數記錄它的出現次數,然後按照出現次數從大到小排序輸出就好了,主要是實現方法,這裡我用了的是map+vector+pair,用時已經是我最快的了,那些更快的就看不太懂他們的代碼了…


AC代碼:

class Solution {  public:      vector<int> topKFrequent(vector<int>& nums, int k) {          vector<int> ans;          vector<pair<int,int>> v;          unordered_map<int,int> ma;          for(int i=0;i<nums.size();i++){              ma[nums[i]] ++;          }          for(auto i : ma){              v.push_back(make_pair(i.first, i.second));          }          sort(v.begin(), v.end(), [](const pair<int,int> a,const pair<int,int> b){              if(a.second == b.second)return a.first < b.first;              return a.second > b.second;          });          for(int i=0;i<k;i++){              ans.push_back(v[i].first);          }          return ans;      }  };