Enable Javascript in your browser and then refresh this page, for a much enhanced experience.
dfs solution in Clear category for What Is Wrong With This Family? by spark856
def is_family(tree):
family = {}
for i,j in tree:
if j in family: return False
family[j] = i
def dfs(n,ok):
if n not in family:return n
parent = family[n]
if parent in ok: return n
ok.append(parent)
return dfs(parent,ok)
parents = {dfs(i,[]) for i in family}
return len(parents)==1
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'
print("Looks like you know everything. It is time for 'Check'!")
April 15, 2020
Comments:
