博客
关于我
【Lintcode】1562. Number of Restaurants
阅读量:227 次
发布时间:2019-02-28

本文共 1236 字,大约阅读时间需要 4 分钟。

给定一个长 l 的列表,里面含的是各个餐馆的坐标。顾客的坐标是 (0, 0)。给定一个正整数 n,先寻找离顾客最近的 n 个餐馆,设这 n 个餐馆离顾客的最远距离是 d,然后再按照列表原来的顺序找到前 n 个距离小于等于 d 的餐馆坐标,将这些坐标返回。如果不足 n 个则返回空列表。这里的距离取欧几里得距离。

首先,将原列表深拷贝一份,然后对新列表排序,找到第 n 个餐馆与顾客的距离 d。然后再遍历原列表,找到前 n 个距离也小于等于 d 的餐馆坐标即可。代码如下:

import java.util.ArrayList;import java.util.List;public class Solution {    public List< List< Integer>> nearestRestaurant(List< List< Integer>> restaurant, int n) {        if (restaurant.size() == n) {            return restaurant;        }        if (restaurant.size() < n) {            return new ArrayList<>();        }        List< List< Integer>> res = new ArrayList<>(restaurant);        res.sort((r1, r2) -> Integer.compare(disSqr(r1), disSqr(r2)));        List< List< Integer>> farthest = res.get(n - 1);        int farDisSqr = disSqr(farthest);        List< List< Integer>> list = new ArrayList<>();        for (List< Integer> rest : restaurant) {            if (disSqr(rest) <= farDisSqr) {                list.add(rest);            }            if (list.size() == n) {                break;            }        }        return list;    }    private int disSqr(List< Integer> rest) {        int x = rest.get(0), y = rest.get(1);        return x * x + y * y;    }}

时间复杂度 O(l log l),空间 O(l)。

转载地址:http://dwjs.baihongyu.com/

你可能感兴趣的文章
Mysql学习总结(19)——Mysql无法创建外键的原因
查看>>
Mysql学习总结(20)——MySQL数据库优化的最佳实践
查看>>
Mysql学习总结(21)——MySQL数据库常见面试题
查看>>
Mysql学习总结(22)——Mysql数据库中制作千万级测试表
查看>>
Mysql学习总结(23)——MySQL统计函数和分组查询
查看>>
Mysql学习总结(24)——MySQL多表查询合并结果和内连接查询
查看>>
Mysql学习总结(25)——MySQL外连接查询
查看>>
Mysql学习总结(26)——MySQL子查询
查看>>
Mysql学习总结(27)——Mysql数据库字符串函数
查看>>
Mysql学习总结(28)——MySQL建表规范与常见问题
查看>>
Mysql学习总结(29)——MySQL中CHAR和VARCHAR
查看>>
Mysql学习总结(2)——Mysql超详细Window安装教程
查看>>
Mysql学习总结(30)——MySQL 索引详解大全
查看>>
Mysql学习总结(31)——MySql使用建议,尽量避免这些问题
查看>>
Mysql学习总结(33)——阿里云centos配置MySQL主从复制
查看>>
Mysql学习总结(35)——Mysql两千万数据优化及迁移
查看>>
Mysql学习总结(36)——Mysql查询优化
查看>>
Mysql学习总结(37)——Mysql Limit 分页查询优化
查看>>
Mysql学习总结(38)——21条MySql性能优化经验
查看>>
Mysql学习总结(39)——49条MySql语句优化技巧
查看>>