Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- lines = list(open('input.txt').readlines())
- parentsDict = {}
- for line in lines:
- par, kid = line[:-1].split(')')
- parentsDict[kid] = par
- def ancestors(kid):
- if kid not in parentsDict:
- return [] # base case: COM orbits no one
- par = parentsDict[kid]
- return [par] + ancestors(par)
- def main1():
- print(sum(len(ancestors(kid)) for kid in parentsDict))
- def main2():
- x, y = ['SAN', 'YOU']
- [a_x, a_y] = map(ancestors, [x, y])
- print(len(set(a_x) ^ set(a_y)))
- main1()
- main2()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement