Advertisement
Guest User

Problem 1

a guest
Apr 28th, 2017
83
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.56 KB | None | 0 0
  1. #!/usr/bin/env python3
  2.  
  3.  
  4. def old_time(root):
  5.     return root["time"] + sum(old_time(t) for t in root["relies_on"])
  6.  
  7.  
  8. def new_time(root):
  9.     return root["time"] + sum(2*new_time(t) for t in root["relies_on"])
  10.  
  11.  
  12. def main():
  13.     n = int(input())
  14.     tests = [{"time": int(x), "relies_on": []} for x in input().split()]
  15.  
  16.     for t in tests:
  17.         deps = [int(x) for x in input().split()][1:]
  18.         t["relies_on"] = [tests[i-1] for i in deps]
  19.  
  20.     print(old_time(tests[0]))
  21.     print(new_time(tests[0]))
  22.  
  23.  
  24. if __name__ == "__main__":
  25.     main()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement