LeetCode题解(1409):查询带键的排列(Python)

题目:原题链接(中等)

标签:数组、树状数组

解法时间复杂度空间复杂度执行用时
Ans 1 (Python) O ( N l o g N ) O(NlogN) O(NlogN) O ( N ) O(N) O(N)80ms (40.46%)
Ans 2 (Python)
Ans 3 (Python)

解法一:

class BIT:
    def __init__(self, n):
        self.n = n
        self.a = [0] * (n + 1)

    @staticmethod
    def lowbit(x):
        return x & (-x)

    def query(self, x):
        ret = 0
        while x > 0:
            ret += self.a[x]
            x -= BIT.lowbit(x)
        return ret

    def update(self, x, dt):
        while x <= self.n:
            self.a[x] += dt
            x += BIT.lowbit(x)


class Solution:
    def processQueries(self, queries: List[int], m: int) -> List[int]:
        size = len(queries)

        bit = BIT(m + size)

        pos = [0] * (m + 1)
        for i in range(1, m + 1):
            pos[i] = size + i
            bit.update(size + i, 1)

        ans = []
        for i, q in enumerate(queries):
            idx = pos[q]
            bit.update(idx, -1)
            ans.append(bit.query(idx))
            idx = pos[q] = size - i
            bit.update(idx, 1)
        return ans
  • 0
    点赞
  • 0
    评论
  • 0
    收藏
  • 一键三连
    一键三连
  • 扫一扫,分享海报

打赏
文章很值,打赏犒劳作者一下
相关推荐
©️2020 CSDN 皮肤主题: 游动-白 设计师:白松林 返回首页

打赏

长行

你的鼓励将是我创作的最大动力

¥2 ¥4 ¥6 ¥10 ¥20
输入1-500的整数
余额支付 (余额:-- )
扫码支付
扫码支付:¥2
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者