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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
##!/usr/bin/env python
""" https://perlweeklychallenge.org/blog/perl-weekly-challenge-055/
Challenge 1 - Flip binary """
from collections import defaultdict
import sys
class BinFlip:
""" Does the flip according to the excercise """
def __init__(self, to_flip: str):
self.to_flip = to_flip
self.current_flip = ""
self.all_flips = defaultdict(list)
self.max_ones = 0
self.l_count = None
self.r_count = None
def process_all_flips(self):
""" process all possible combinations """
for l_count in range(len(self.to_flip)):
for r_count in range(l_count, len(self.to_flip)):
self.l_count = l_count
self.r_count = r_count
self.current_flip = self.do_flip()
self.process_current()
return self.max_ones, self.all_flips[self.max_ones]
def do_flip(self):
""" do the flip and return the result """
result = list(self.to_flip)
for str_pos in range(self.l_count, self.r_count + 1):
result[str_pos] = "0" if self.to_flip[str_pos] == "1" else "1"
return ''.join(result)
def process_current(self):
""" Process the current flip result """
ones_count = self.current_flip.count("1")
self.all_flips[ones_count].append((self.l_count, self.r_count))
if ones_count > self.max_ones:
self.max_ones = ones_count
class BinFlipTest:
""" run the tests """
def __init__(self):
self.test_flipper = None
def test_all(self):
""" Do the tests """
self.test_flipper = BinFlip("010")
max_ones, all_flips = self.test_flipper.process_all_flips()
assert max_ones == 2
assert all_flips == [(0, 0), (0, 2), (2, 2)]
def main():
""" main """
num_flipper = BinFlip(sys.argv[1])
max_ones, all_flips = num_flipper.process_all_flips()
print(f'Max: {max_ones}, l,r: {all_flips}')
if __name__ == "__main__":
assert len(sys.argv) == 2
main()
flip_tester = BinFlipTest()
flip_tester.test_all()
|
