aboutsummaryrefslogtreecommitdiff
path: root/challenge-046/lubos-kolouch/python/ch-2.py
blob: 0fda3ef76eb3b08c8fe760014b7694cd2e250403 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
#!/usr/bin/env python
# -*- coding: utf-8 -*-


def find_open_rooms() -> [int]:
    rooms_count = 500
    rooms = [False] * (rooms_count + 1)

    for i in range(1, rooms_count + 1):
        for j in range(i, rooms_count + 1, i):
            rooms[j] = not rooms[j]

    open_rooms = [i for i in range(1, rooms_count + 1) if rooms[i]]
    return open_rooms


if __name__ == "__main__":
    open_rooms = find_open_rooms()
    print("Open rooms:", ", ".join(map(str, open_rooms)))