leetcode 973 最接近原点的K个点 快速排序+优先队列+排序+维护 全四种做法

按顺序做好难坚持,决定开始做每日一题了,希望能坚持下去。

解法一:是我一开始想到的解法,就是维护一个长度为K的数组,每个点更新一次。
设点的个数为N,复杂度为O(KN)

from math import sqrt
class Solution:
    def kClosest(self, points: List[List[int]], K: int) -> List[List[int]]:
        dists, result = [], []
        for point in points:
            a, b = point
            dist = sqrt(a*a + b*b)
            if len(dists) > 0 and dist < dists[-1]:
                for i in range(len(dists)):
                    if dist < dists[i]:
                        dists.insert(i, dist)
                        result.insert(i, point)
                        break
            elif len(dists) < K:
                dists.append(dist)
                result.append(point)
            if len(dists) > K:
                result.pop()
                dists.pop()
        return result

解法二:优先队列
维护一个有序小数组,最应该想到的就是优先队列。。草率了。堆还是用c++更舒服一点
另外,用整数存距离(即不做平方根)要比浮点数更快一些。
此算法可以实现O(NlogK)复杂度,在每一步push加pop就行,然而在这个题中,由于数据量小,时间要求严格,不加pop也是O(NlogK),其实加pop才是严格的最小时间复杂度。

class Solution {
   
public:
    struct node
    {
   
        vector<int> point;
        int dist;

        node(vector<int> point, int dist)
        {
   
            this->point = point;
            this->dist = dist;
        }
    };

    struct cmp
    {
   
        bool operator () (node a, node b)
        {
   
            return a.dist > b.dist;
        }
    };

    vector<vector
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值