題目地址：

https://www.lintcode.com/problem/number-of-restaurants/description

給定一個長$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 {
    /**
     * @param restaurant:
     * @param n:
     * @return: nothing
     */
    public List<List<Integer>> nearestRestaurant(List<List<Integer>> restaurant, int n) {
        // Write your code here
        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<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)$。