Enable Javascript in your browser and then refresh this page, for a much enhanced experience.
Collections defaultdict solution in Clear category for Friends by swagg010164
from collections import defaultdict
class Friends:
def __init__(self, connections):
self.connections = defaultdict(set)
for k, v in connections:
self.__add(k, v)
def __add(self, key, value):
self.connections[key].add(value)
self.connections[value].add(key)
def add(self, connection):
k, v = connection
if v in self.connections[k]:
return False
self.__add(k, v)
return True
def remove(self, connection):
k, v = connection
if k in self.connections:
self.connections[k].discard(v)
self.connections[v].discard(k)
return True
return False
def names(self):
return set(k for k in self.connections if self.connections[k])
def connected(self, name):
return self.connections[name]
if __name__ == '__main__':
#These "asserts" using only for self-checking and not necessary for auto-testing
letter_friends = Friends(({"a", "b"}, {"b", "c"}, {"c", "a"}, {"a", "c"}))
digit_friends = Friends([{"1", "2"}, {"3", "1"}])
assert letter_friends.add({"c", "d"}) is True, "Add"
assert letter_friends.add({"c", "d"}) is False, "Add again"
assert letter_friends.remove({"c", "d"}) is True, "Remove"
assert digit_friends.remove({"c", "d"}) is False, "Remove non exists"
assert letter_friends.names() == {"a", "b", "c"}, "Names"
assert letter_friends.connected("d") == set(), "Non connected name"
assert letter_friends.connected("a") == {"b", "c"}, "Connected name"
July 9, 2020
Comments: