Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- def f(lst):
- nums = {}
- n, nm = None, 0
- for i in lst:
- nums[i] = t = nums.get(i, 0) + 1
- if t > nm:
- n, nm = i, t
- return n
- N = int(input())
- roads = []
- if N == 1:
- print(0, 0)
- quit()
- for i in range(N - 1):
- roads.append(input().split())
- # val = list(itertools.chain.from_iterable(roads))
- val = [j for sub in roads for j in sub]
- city = f(val)
- count = 0
- for i in range(N - 1):
- for g in range(2):
- if int(roads[i][g]) == int(city):
- count += 1 # TODO rewrite
- a = 1
- c = N - 1 - count
- if c == 0:
- print(count, city)
- quit()
- else:
- for i in range(c):
- count += (a + 1) ** 2
- a += 1
- print(count, city)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement