forked from doocs/leetcode
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Solution2.py
39 lines (33 loc) · 954 Bytes
/
Solution2.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
class BinaryIndexedTree:
def __init__(self, n):
self.n = n
self.c = [0] * (n + 1)
@staticmethod
def lowbit(x):
return x & -x
def update(self, x, delta):
while x <= self.n:
self.c[x] += delta
x += BinaryIndexedTree.lowbit(x)
def query(self, x):
s = 0
while x > 0:
s += self.c[x]
x -= BinaryIndexedTree.lowbit(x)
return s
class Solution:
def processQueries(self, queries: List[int], m: int) -> List[int]:
n = len(queries)
pos = [0] * (m + 1)
tree = BinaryIndexedTree(m + n)
for i in range(1, m + 1):
pos[i] = n + i
tree.update(n + i, 1)
ans = []
for i, v in enumerate(queries):
j = pos[v]
tree.update(j, -1)
ans.append(tree.query(j))
pos[v] = n - i
tree.update(n - i, 1)
return ans