一亩三分地论坛

 找回密码
 获取更多干货,去instant注册!

扫码关注一亩三分地公众号
查看: 252|回复: 0
收起左侧

[算法题] top K 问题C++ 版本解法

[复制链接] |试试Instant~ |关注本帖
thewave 发表于 2015-11-5 01:45:02 | 显示全部楼层 |阅读模式

注册一亩三分地论坛,查看更多干货!

您需要 登录 才可以下载或查看,没有帐号?获取更多干货,去instant注册!

x
Amazon最近的OA经常有这个题。
网上比较多是Java版本,我写了个C++。
  1. struct myComp {
  2.         bool operator()(const pair<int, int> &a, const pair<int, int> &b) {
  3.             return a.first * a.first + a.second * a.second < b.first * b.first + b.second * b.second;
  4.         }
  5.     };

  6.     int calDistance(const pair<int, int> &a) {
  7.         return a.first * a.first + a.second * a.second;
  8.     }

  9.     vector<pair<int, int>> topK(vector<pair<int, int>> nums, int k) {
  10.         int n = nums.size();
  11.         if (k >= n)return nums;
  12.         priority_queue<pair<int, int>, vector<pair<int, int>>, myComp> heap;
  13.         for (int i = 0; i < k; i++) {
  14.             heap.push(nums[i]);
  15.         }
  16.         for (int i = k; i < n; i++) {
  17.             if (calDistance(nums[i]) < calDistance(heap.top())) {
  18.                 heap.pop();
  19.                 heap.push(nums[i]);
  20.             }
  21.         }

  22.         vector<pair<int, int>> res;
  23.         for (int i = 0; i < k; i++) {
  24.             res.push_back(heap.top());
  25.             heap.pop();
  26.         }
  27.         return res;

  28.     }
复制代码

本版积分规则

请点这里访问我们的新网站:一亩三分地Instant.

Instant搜索更强大,不扣积分,内容组织的更好更整洁!目前仍在beta版本,努力完善中!反馈请点这里

关闭

一亩三分地推荐上一条 /5 下一条

手机版|小黑屋|一亩三分地论坛声明 ( 沪ICP备11015994号 )

custom counter

GMT+8, 2016-12-9 10:30

Powered by Discuz! X3

© 2001-2013 Comsenz Inc. Design By HUXTeam

快速回复 返回顶部 返回列表