Skip to content

Commit 61d3a55

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent c23f6b5 commit 61d3a55

File tree

1 file changed

+17
-11
lines changed

1 file changed

+17
-11
lines changed

data_structures/trie/binary_trie.py

+17-11
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ def __init__(self, max_bit_len=31):
88
def add(self, a):
99
u = 0
1010
self.cc[u] += 1
11-
for i in range(self.mb-1, -1, -1):
11+
for i in range(self.mb - 1, -1, -1):
1212
d = a >> i & 1
1313
if self.to[d][u] == -1:
1414
self.to[d][u] = len(self.cc)
@@ -19,29 +19,34 @@ def add(self, a):
1919
self.cc[u] += 1
2020

2121
def remove(self, a):
22-
if self.cc[0] == 0: return False
22+
if self.cc[0] == 0:
23+
return False
2324
uu = [0]
2425
u = 0
25-
for i in range(self.mb-1, -1, -1):
26+
for i in range(self.mb - 1, -1, -1):
2627
d = a >> i & 1
2728
u = self.to[d][u]
28-
if u == -1 or self.cc[u] == 0: return False
29+
if u == -1 or self.cc[u] == 0:
30+
return False
2931
uu.append(u)
30-
for u in uu: self.cc[u] -= 1
32+
for u in uu:
33+
self.cc[u] -= 1
3134
return True
3235

3336
def cnt(self, a):
3437
u = 0
35-
for i in range(self.mb-1, -1, -1):
38+
for i in range(self.mb - 1, -1, -1):
3639
d = a >> i & 1
3740
u = self.to[d][u]
38-
if u == -1 or self.cc[u] == 0: return 0
41+
if u == -1 or self.cc[u] == 0:
42+
return 0
3943
return self.cc[u]
4044

4145
def min_xor(self, a):
42-
if self.cc[0] == 0: return self.inf
46+
if self.cc[0] == 0:
47+
return self.inf
4348
u, res = 0, 0
44-
for i in range(self.mb-1, -1, -1):
49+
for i in range(self.mb - 1, -1, -1):
4550
d = a >> i & 1
4651
v = self.to[d][u]
4752
if v == -1 or self.cc[v] == 0:
@@ -52,9 +57,10 @@ def min_xor(self, a):
5257
return res
5358

5459
def max_xor(self, a):
55-
if self.cc[0] == 0: return -self.inf
60+
if self.cc[0] == 0:
61+
return -self.inf
5662
u, res = 0, 0
57-
for i in range(self.mb-1, -1, -1):
63+
for i in range(self.mb - 1, -1, -1):
5864
d = a >> i & 1
5965
v = self.to[d ^ 1][u]
6066
if v == -1 or self.cc[v] == 0:

0 commit comments

Comments
 (0)