买房小白任秀坡在湾区买房经历(一)

一亩三分地论坛

 找回密码
 Sign Up 注册获取更多干货
码农求职神器Triplebyte:
不用海投,内推多家公司面试
国内机会:实时大数据分析领域领导者
Web/大数据/机器学习/产品等职位
日志易机器大数据行业践行者Web/大数据/机器学习等职位-北京or深圳
把贵司招聘信息放这里
查看: 595|回复: 0
收起左侧

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

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

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

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

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.     }
复制代码

本版积分规则

提醒:发帖可以选择内容隐藏,部分板块支持匿名发帖。请认真读完以下全部说明:

■隐藏内容方法: [hide=200]你想要隐藏的内容比如面经[/hide]
■意思是:用户积分低于200则看不到被隐藏的内容
■可以自行设置积分值,不建议太高(200以上太多人看不到),也不建议太低(那就没必要隐藏了)
■建议只隐藏关键内容,比如具体的面试题目、涉及隐私的信息,大部分内容没必要隐藏。
■微信/QQ/电子邮件等,为防止将来被骚扰甚至人肉,以论坛私信方式发给对方最安全。
■匿名发帖的板块和方法:http://www.1point3acres.com/bbs/thread-388663-1-1.html

关闭

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

手机版|小黑屋|一亩三分地论坛声明

custom counter

GMT+8, 2018-4-19 19:56

Powered by Discuz! X3

© 2001-2013 Comsenz Inc. Design By HUXTeam

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