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

最新推荐文章于 2024-09-18 11:04:20 发布

长行

最新推荐文章于 2024-09-18 11:04:20 发布

阅读量232

点赞数

分类专栏：刷题记录文章标签： leetcode python 数组树状数组

本文链接：https://blog.csdn.net/Changxing_J/article/details/112869449

版权

刷题记录专栏收录该内容

1966 篇文章 55 订阅

订阅专栏

题目：原题链接（中等）

标签：数组、树状数组

解法	时间复杂度	空间复杂度	执行用时
Ans 1 (Python)	$O (N l o g 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