973. 最接近原点的 K 个点

lankerens

于 2020-11-09 19:02:11 发布

阅读量101

点赞数

分类专栏： LeetCodeDaily 文章标签： hashmap 哈希表 java leetcode

本文链接：https://blog.csdn.net/qq_43765535/article/details/109583580

版权

LeetCodeDaily 专栏收录该内容

287 篇文章

订阅专栏

LeetCode： 973. 最接近原点的 K 个点

在这里插入图片描述

哈希表暴力 >> 或者用大顶堆

map暴力



    public int[][] kClosest(int[][] points, int K) {
        // 默认按 key 的升序
        Map<Double, List<Integer>> map = new TreeMap<>();

        // 初始化 map
        for (int i = 0; i < points.length; i++) {
            double dis = Math.sqrt(Math.pow(points[i][0], 2) + Math.pow(points[i][1], 2));
            List<Integer> list = map.get(dis);
            if(list == null) list = new ArrayList<>();
            list.add(i);
            map.put(dis, list);
        }

        int[][] ans = new int[K][2];
        int cnt = 0;
        // 遍历
        for (List<Integer> val: map.values()) {
            for (Integer index: val) {
                ans[cnt++] = points[index];
                K--;
                if(K == 0) break;
            }
            if(K == 0) break;
        }

        return ans;
    }