Enable Javascript in your browser and then refresh this page, for a much enhanced experience.
Concise algorithm solution in Speedy category for What Is Wrong With This Family? by Igor_Sekretarev
from collections import defaultdict
def is_family(tree: list[list[str]]) -> bool:
ancestors = defaultdict(set)
for father, son in tree:
if ( father == son or ancestors[son] or son in ancestors[father]
or ancestors[father] & ancestors[son]
):
return False
ancestors[son] = {father} | ancestors[father]
roots = 0
for man in ancestors:
if not ancestors[man]:
roots += 1
if roots > 1:
return False
return True
if __name__ == "__main__":
#These "asserts" using only for self-checking and not necessary for auto-testing
assert is_family([
['Logan', 'Mike']
]) == True, 'One father, one son'
assert is_family([
['Logan', 'Mike'],
['Logan', 'Jack']
]) == True, 'Two sons'
assert is_family([
['Logan', 'Mike'],
['Logan', 'Jack'],
['Mike', 'Alexander']
]) == True, 'Grandfather'
assert is_family([
['Logan', 'Mike'],
['Logan', 'Jack'],
['Mike', 'Logan']
]) == False, 'Can you be a father to your father?'
assert is_family([
['Logan', 'Mike'],
['Logan', 'Jack'],
['Mike', 'Jack']
]) == False, 'Can you be a father to your brother?'
assert is_family([
['Logan', 'William'],
['Logan', 'Jack'],
['Mike', 'Alexander']
]) == False, 'Looks like Mike is stranger in Logan\'s family'
assert is_family([
['Jack', 'Mike'],
['Logan', 'Mike'],
['Logan', 'Jack'],
]) == False, 'Two fathers'
print("Looks like you know everything. It is time for 'Check'!")
April 25, 2021
